Partilhar via

Como instalar o suporte a GPU no Construtor de Modelos

  • Artigo

Saiba como instalar os drivers da GPU para usar sua GPU com o Model Builder.

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 6GB de memória GPU dedicada.

Pré-requisitos

Apenas classificação de imagens

  • Conta de desenvolvedor NVIDIA. Se não tiver uma subscrição, crie uma conta gratuita.
  • Instalar dependências
    • Instale CUDA v10.1. Certifique-se de instalar o CUDA v10.1, não qualquer outra versão mais recente.
    • Instale cuDNN v7.6.4 para CUDA 10.1. Não é possível 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.

Resolução de Problemas

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

Os cenários de aprendizagem profunda tendem a ser executados mais rapidamente em GPUs.

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

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

Como sei qual GPU tenho?

  1. Clique com o botão direito do rato no ambiente de trabalho
  2. Se vir "NVIDIA Control Panel" ou "NVIDIA Display" na janela pop-up, tem um GPU NVIDIA
  3. Clique em "NVIDIA Control Panel" ou "NVIDIA Display" na janela pop-up
  4. Olhe para "Informações da placa gráfica"
  5. Você verá o nome da sua GPU NVIDIA

Eu não vejo o Painel de Controle da NVIDIA (ou ele não abre), mas sei que tenho uma GPU NVIDIA.

  1. Abra o Gestor de Dispositivos
  2. Olhe para adaptadores de vídeo
  3. Instale o driver apropriado para sua GPU.

Como posso ver a versão do CUDA que tenho?

  1. Abrir uma janela do PowerShell ou da linha de comando
  2. Digite nvcc --version

CUDA não está disponível, confirme que você tem uma GPU suportada pelo 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 as atualizações, você pode obter os drivers mais recentes de https://www.nvidia.com/geforce/drivers/.
  3. Instale os drivers mais recentes.