Consulta TP 2: Usar GPU para entrenar

22 views
Skip to first unread message

Mateo Tomas

unread,
Nov 1, 2024, 10:51:39 AM11/1/24
to UCSE IA
Buenas profes
Molesto con una duda, tengo instalado Tensorflow  2.18.0 con cuDNN 9.3 y CUDA 12.5 en windows 11 con una placa NVIDIA 4050
Sin embargo al correr la linea 
print("GPUs disponibles:", len(tf.config.list_physical_devices('GPU')))

No me devuelve ningun resultado de GPU detectada. Vale aclarar que ya tengo instalados los ultimos drivers.

Falta hacer una configuracion extra ademas de instalar CUDA y cuDNN? Tambien agregue a las variables de entorno las rutas a ambos directorios 

Saludos.

Juan Carignano

unread,
Nov 1, 2024, 11:10:35 AM11/1/24
to UCSE IA
Buenas

Si bien le preguntaste a los profes, te comento que probé de todo y lo único que me sirvió fue seguir al pie de la letra este video https://www.youtube.com/watch?v=FZIuiIE28NY. Una vez creado el entorno, solamente tuve que cambiar el kernel y pude usar mi GPU. Si llegas a probarlo, espero que te sirva.

Saludos

fisa

unread,
Nov 1, 2024, 11:17:15 AM11/1/24
to ucs...@googlegroups.com
Buenas!

Mirando la doc de instalación de TensorFlow ( https://www.tensorflow.org/install/pip#windows-wsl2_1 ) veo algo importante: en teoría la versión 2.10 fue la última en soportar GPUs de forma nativa en Windows. A partir de la 2.11 en Windows hay que usar WSL2 para poder correr con GPU. WSL2 es básicamente un "Linux adentro de Windows" (Windows es tan bueno que Microsoft tuvo que meterle un Linux adentro :p), y requiere algunos pasos extras para instalar cosas que están detallados en esa página.

Así que tenés dos opciones:

Opción A) Usar una versión más vieja de TensorFlow que todavía tenga soporte nativo de GPU en windows. En PyPI veo que el paquete tensorflow-gpu tiene hasta la versión 2.12. Probaría esa, si no la 2.10. El video que compartió Juan hace básicamente eso, pero con versiones demasiaaaaado viejas (de hace 4 años). Te recomendaría probar con versiones más recientes.

Opción B) Usar WSL2 para poder correr las versiones más nuevas de TensorFlow con GPU. Nunca lo hice, con lo que no te puedo garantizar que sea simple, pero a primera vista parece que implica: instalar WSL2, luego instalar y configurar CUDA adentro de WSL, y finalmente un "pip install tensorflow[and-cuda]" (y empezar a correr siempre todo dentro de WSL2).

Espero que esto ayude. Saludos!


--
You received this message because you are subscribed to the Google Groups "UCSE IA" group.
To unsubscribe from this group and stop receiving emails from it, send an email to ucse-ia+u...@googlegroups.com.
To view this discussion visit https://groups.google.com/d/msgid/ucse-ia/a57ca31b-4adb-403b-b383-43516d35c108n%40googlegroups.com.


--
fisa  -  Juan Pedro Fisanotti
Reply all
Reply to author
Forward
0 new messages