¿Qué debemos saber para instalar correctamente TensorFlow con GPU?
Hacer uso de una GPU para entrenar nuestros modelos de Deep Learning es algo casi imprescindible hoy en día, por lo “rápido” que resulta entrenar los modelos con GPU respecto a entrenarlos con una CPU. En el siguiente enlace tenéis la explicación del porqué la GPU es más rápida que la CPU para entrenar los modelos de Deep Learning: Python vs Numpy vs GPU (TensorFlow): Tiempos de Ejecución
Si es la primera vez que te enfrentas a la instalación de TensorFlow y quieres hacer uso de la GPU, tienes que saber qué debes instalar en tu PC (drivers de NVidia, CUDA, cuDNN, Python y TensorFlow-gpu) y con que versiones, para que TensorFlow haga uso de la GPU. En el siguiente video lo contamos con todo detalle:
En resumen, debes de responder a las siguientes preguntas (ENLACES):
- ¿Qué sistema operativo utilizo?
- ¿Qué GPU tengo?
- ¿Qué versión de drivers me puedo instalar?
- Descarga de Drivers: https://www.nvidia.es/Download/index.aspx?lang=es
- ¿Qué versiones de CUDA me puedo instalar en función de la versión de los Drivers ya instalados?
- Compatibilidades CUDA vs Drivers: https://docs.nvidia.com/deploy/cuda-compatibility/index.html#minor-version-compatibility
- Descarga de CUDA: https://developer.nvidia.com/cuda-downloads
- ¿Qué versiones de TensorFlow (con Python) puedo instalarme en función de la versión de CUDA? Y ¿Qué versión de cuDNN me instalo?
- Compatibilidades Python vs TensorFlow vs CUDA vs cuDNN: https://www.tensorflow.org/install/source_windows#gpu
- Descarga de cuDNN: https://developer.nvidia.com/rdp/cudnn-archive (debes de registrarte para poder descargar cuDNN)
