Saltar al contenido

¿Cómo uso la GPU TensorFlow?

La guía o código que verás en este post es la solución más eficiente y efectiva que hallamos a tu duda o problema.

Solución:

Sigue este tutorial GPU Tensorflow lo hice y funciona perfecto.

¡Atención! – Instalar en pc versión 9.0! la versión más nueva no es compatible con Tensorflow-gpu

Pasos:

  1. Desinstala tu antiguo tensorflow
  2. Instalar tensorflow-gpu pip install tensorflow-gpu
  3. Instale la tarjeta gráfica y los controladores de Nvidia (probablemente ya los tenga)
  4. Descargar e instalar CUDA
  5. Descargar e instalar cuDNN
  6. Verificar por programa simple

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

Primero necesitas instalar tensorflow-gpu, porque este paquete es responsable de los cálculos de gpu. También recuerde ejecutar su código con la variable de entorno. CUDA_VISIBLE_DEVICES = 0 (o si tiene varias gpus, ponga sus índices con coma). Puede haber algunos problemas relacionados con el uso de gpu. si su tensorflow no usa gpu de todos modos, intente esto

La ‘nueva’ forma de instalar GPU tensorflow si tiene Nvidia, es con Anaconda. Funciona en Windows también. Con 1 línea.

conda create --name tf_gpu tensorflow-gpu 

Este es un atajo para 3 comandos, que puede ejecutar por separado si lo desea o si ya tiene un entorno conda y no necesita crear uno.

  1. Crear un entorno de anaconda conda create --name tf_gpu

  2. Activa el entorno activate tf_gpu

  3. Instalar tensorflow-GPU conda install tensorflow-gpu

Puede utilizar el entorno conda.

Si posees algún atolladero y disposición de avanzar nuestro sección eres capaz de añadir una disquisición y con mucho placer lo interpretaremos.

¡Haz clic para puntuar esta entrada!
(Votos: 0 Promedio: 0)



Utiliza Nuestro Buscador

Deja una respuesta

Tu dirección de correo electrónico no será publicada. Los campos obligatorios están marcados con *