Como instalar o suporte à GPU no Construtor de Modelos

  • Artigo

Saiba como instalar os drivers de GPU para usar sua GPU com o Construtor de Modelos.

Requisitos de hardware

  • Pelo menos uma GPU compatível com CUDA. Para obter uma lista de GPUs compatíveis, consulte o guia da NVIDIA.
  • Pelo menos 6 GB de memória de GPU dedicada.

Pré-requisitos

Somente classificação de imagem

  • Conta de desenvolvedor NVIDIA. Se você não tiver uma, crie uma conta gratuita.
  • Instalar dependências
    • Instale o CUDA v10.1. Instale o CUDA v10.1, não qualquer outra versão mais recente.
    • Instale cuDNN v7.6.4 para CUDA 10.1. Você não pode ter várias versões do cuDNN instaladas. Depois de baixar o arquivo zip cuDNN v7.6.4 e descompactá-lo, copie <CUDNN_zip_files_path>\cuda\bin\cudnn64_7.dll para <YOUR_DRIVE>\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin.

Solução de problemas

E se eu não tiver uma GPU instalada localmente?

Cenários de aprendizado profundo tendem a ser executados mais rapidamente em GPUs.

Alguns cenários, como a classificação de imagem, dão suporte ao treinamento em VMs de GPU do Azure.

No entanto, se GPUs locais ou o Azure não são uma opção para você, esses cenários também são executados na CPU. No entanto, observe que os tempos de treinamento são significativamente mais longos.

Como saber qual GPU eu tenho?

  1. Clique com o botão direito do mouse na área de trabalho
  2. Se você vir "NVIDIA Painel de Controle" ou "NVIDIA Display" na janela pop-up, terá uma GPU NVIDIA
  3. Clique em "NVIDIA Painel de Controle" ou "NVIDIA Display" na janela pop-up
  4. Veja "Informações da Placa Gráfica"
  5. Você verá o nome da GPU NVIDIA

Não vejo a NVIDIA Painel de Controle (ou falha ao abrir), mas sei que tenho uma GPU NVIDIA.

  1. Abra o Gerenciador de Dispositivos
  2. Examinar adaptadores de exibição
  3. Instale o driver apropriado para sua GPU.

Como ver qual versão de CUDA eu tenho?

  1. Abra um PowerShell ou uma janela de linha de comando
  2. Digite nvcc --version

cuda não está disponível, confirme se você tem uma GPU compatível com cuda

  1. Abra o aplicativo GeForce Experience.
  2. O aplicativo deve mostrar atualizações de driver instaladas e disponíveis. Se você tiver problemas para ver atualizações, poderá obter os drivers mais recentes de https://www.nvidia.com/geforce/drivers/.
  3. Instale os drivers mais recentes.