puigalex / AMP-Tech

106 stars 340 forks source link

Tensorflow-gpu #6

Open SERIBO92 opened 4 years ago

SERIBO92 commented 4 years ago

Hola, sucede que he estado intentando entrenar una CNN y ya instale el tensorflow-gpu, sin embargo al correr los codigos y revisar, me aparece el uso en cero siempre, como en las imagenes. En este caso estoy usando el codigo de clasificacion de animales de tu canal de youtube (perro, gato y gorila), no sé si de pronto conozcas cual puede ser el inconveniente o si de pronto es que estoy haciendo algo mal. Gracias

1 2 3 4

puigalex commented 4 years ago

Hola,

¿Instalaste los drivers necesarios para que funcione Tensorflow-gpu? CuDNN y Cuda Toolkit?

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html

https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html