Instale la última versión de TensorFlow que admita CUDA, cudNN y GPU en una PC con Windows 11

introducir

Si eres como yo, sabes que configurar TensorFlow en un dispositivo nuevo puede ser complicado. En este tutorial, cubriré la instalación de CUDA, cuDNN y TensorFlow en Windows 11. El objetivo de este artículo es hacer que ese proceso sea lo menos doloroso posible. Este artículo supone que ya tienes una GPU compatible con CUDA instalada en tu PC.

Pasos de instalación

Es importante seguir cada paso en orden. Si realiza los pasos fuera de orden, tendrá que comenzar el proceso de nuevo, lo cual nadie quiere hacer. Entonces, sin más preámbulos, vayamos directo al grano.

Software que necesitamos instalar:

Microsoft Visual Studio (Edición comunitaria 2019)
NVIDIA CUDA Toolkit
NVIDIA cuDNN
Python (yo personalmente uso anaconda)
TensorFlow

Paso 1: elimine todos los controladores NVIDIA actuales

Vaya a la barra de búsqueda en Windows y busque "Agregar o quitar programas". Ahora debería ver una lista de aplicaciones instaladas. Desplázate hacia abajo hasta que veas algo que diga NVIDIA. Debería verse similar a la imagen a continuación.

inserte la descripción de la imagen aquí
Ahora verifique y elimine todo lo que esté marcado como NVIDIA. No se preocupe por eliminarlos, los instalaremos nuevamente en un paso posterior.

Paso 2: instale Visual Studio (no el código VS)

La instalación del kit de herramientas CUDA requiere Visual Studio. Si intentas desproteger

Supongo que te gusta

Origin blog.csdn.net/iCloudEnd/article/details/132423014
Recomendado
Clasificación