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 guia da NVIDIA.
  • Pelo menos 6 GB de memória GPU dedicada.

Pré-requisitos

Apenas classificação de imagens

  • Conta de desenvolvedor NVIDIA. Se não tiveres uma, cria 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 a partir do arquivo cuDNN. 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.

Soluçã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. No entanto, os tempos de treino são significativamente mais longos.

Como sei qual GPU tenho?

Verifique a GPU na Configurações

  1. Clique com o botão direito do mouse no ícone do menu Iniciar do Windows e selecione Configurações.
  2. Selecione Configurações>Sistema
  3. Selecione Ecrã e desça até Configurações Relacionadas.
  4. Selecione Exibição avançada. A marca e o modelo da GPU são apresentados em informações de visualização.

Verificar GPU a partir do Gestor de Tarefas

  1. Clique com o botão direito do rato no ícone do menu Iniciar do Windows e selecione Gestor de Tarefas.
  2. Selecione de desempenho.
  3. No último painel da guia, escolha GPU. Se esta opção estiver disponível, provavelmente estará na parte inferior da lista.
  4. No canto superior direito da seleção de GPU, são mostradas informações sobre a GPU do seu computador.

Não vejo a minha GPU em Definições ou no Gestor de Tarefas, mas sei que tenho uma GPU NVIDIA.

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

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

  1. Abra uma janela do PowerShell ou da linha de comando.
  2. Execute o comando nvcc --version.

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