Compartilhar via

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?

Verificar 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 sua GPU devem aparecer em Informações da tela.

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 GPUs, as informações sobre a GPU do seu computador estarão visíveis.

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