Compartilhar via

Como instalar o suporte à GPU no Construtor de Modelos

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 GPU dedicada.

Pré-requisitos

Somente classificação de imagem

  • Conta de desenvolvedor NVIDIA. Se você não tiver uma, crie uma conta gratuita.
  • Instale as dependências:
    • Instalar 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 do arquivo de 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?

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, os tempos de treinamento são significativamente mais longos.

Como saber qual GPU eu tenho?

Verifique a GPU nas 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 Tela e role para baixo até Configurações relacionadas.
  4. Selecione Exibição avançada. A marca e o modelo da GPU são mostrados em Informações de exibição.

Verificar a GPU no Gerenciador de Tarefas

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

Não estou vendo minha GPU em Configurações nem no Gerenciador de Tarefas, mas sei que tenho uma GPU NVIDIA.

  1. Abra o Gerenciador de Dispositivos.
  2. Examine 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. Execute o comando 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.