Comparteix a través de

Procedimiento para instalar la compatibilidad con GPU en Model Builder

  • Article

Aprenda a instalar los controladores de GPU para usar la GPU con Model Builder.

Requisitos de hardware

  • Al menos una GPU compatible con CUDA. Para obtener una lista de las GPU compatibles, vea la guía de NVIDIA.
  • Al menos 6 GB de memoria de GPU dedicada.

Prerrequisitos

Solo clasificación de imágenes

  • Cuenta de desarrollador de NVIDIA. En caso de no tener ninguna, cree una cuenta gratuita.
  • Instalación de dependencias
    • Instale CUDA v10.1. Asegúrese de instalar CUDA v10.1, y no cualquier otra versión más reciente.
    • Instale cuDNN v7.6.4 para CUDA 10.1. No puede tener varias versiones de cuDNN instaladas. Después de descargar el archivo ZIP de cuDNN v7.6.4 y desempaquetarlo, copie <CUDNN_zip_files_path>\cuda\bin\cudnn64_7.dll en <YOUR_DRIVE>\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin.

Solución de problemas

¿Qué ocurre si no tengo una GPU instalada en el entorno local?

Los escenarios de aprendizaje profundo tienden a ejecutarse más rápido en las GPU.

Algunos escenarios, como la clasificación de imágenes, admiten el entrenamiento en máquinas virtuales de GPU de Azure.

Sin embargo, si las GPU locales o Azure no son una opción para usted, estos escenarios también se ejecutan en la CPU. Tenga en cuenta que los tiempos de entrenamiento son significativamente más largos.

¿Cómo puedo saber qué GPU tengo?

Comprobación de la GPU desde Configuración

  1. Haga clic con el botón derecho en el icono del menú Inicio de Windows y seleccione Configuración.
  2. Seleccione Configuración>Sistema.
  3. Seleccione Mostrar y desplácese hacia abajo hasta Configuración relacionada.
  4. Seleccione Pantalla avanzada. La marca y el modelo de la GPU deben mostrarse en Mostrar información.

Comprobación de la GPU desde el Administrador de tareas

  1. Haga clic con el botón derecho en el icono del menú Inicio de Windows y seleccione Administrador de tareas.
  2. Seleccione Rendimiento.
  3. En el último panel de la pestaña, elija GPU. Si esta opción está disponible, es probable que esté en la parte inferior de la lista.
  4. En la esquina superior derecha de la selección de GPU, se mostrará información sobre la GPU del equipo.

No veo mi GPU en Configuración ni en el Administrador de tareas, pero sé que tengo una GPU NVIDIA.

  1. Abra el Administrador de dispositivos
  2. Consulte los adaptadores de pantalla.
  3. Instale el controlador adecuado para la GPU.

¿Cómo puedo ver qué versión de CUDA tengo?

  1. Abra una ventana de PowerShell o de línea de comandos
  2. Tipo en nvcc --version

cuda no está disponible, confirme que tiene una GPU compatible con cuda.

  1. Abra la aplicación GeForce Experience.
  2. La aplicación debe mostrar las actualizaciones de los controladores instaladas y disponibles. Si tiene problemas para ver las actualizaciones, en https://www.nvidia.com/geforce/drivers/ encontrará los controladores más recientes.
  3. Instale los controladores más recientes.