¿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):

  1. ¿Qué sistema operativo utilizo?
  2. ¿Qué GPU tengo?
  3. ¿Qué versión de drivers me puedo instalar?
  4. ¿Qué versiones de CUDA me puedo instalar en función de la versión de los Drivers ya instalados?
  5. ¿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?
Comparte esta entrada en:
Safe Creative #1401310112503
¿Qué debemos saber para instalar correctamente TensorFlow con GPU? por "www.jarroba.com" esta bajo una licencia Creative Commons
Reconocimiento-NoComercial-CompartirIgual 3.0 Unported License.
Creado a partir de la obra en www.jarroba.com

Deja una respuesta



Tu dirección de correo electrónico no será publicada.

Este sitio usa Akismet para reducir el spam. Aprende cómo se procesan los datos de tus comentarios.

Uso de cookies

Este sitio web utiliza cookies para que usted tenga la mejor experiencia de usuario. Si continúa navegando está dando su consentimiento para la aceptación de las mencionadas cookies y la aceptación de nuestra política de cookies

ACEPTAR
Aviso de cookies