Como instalar o suporte a GPU no Construtor de Modelos
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
- Extensão do Visual Studio do Construtor de Modelos. A extensão é incorporada no Visual Studio a partir da versão 16.6.1.
- Verifique se o driver apropriado está instalado para a GPU.
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?
Verifique a GPU em Configurações
- Clique com o botão direito do mouse no ícone do menu Iniciar do Windows e selecione Configurações.
- Selecione Configurações>do sistema
- Selecione Exibir e role para baixo até Configurações relacionadas.
- Selecione Visualização avançada. A marca e o modelo da GPU devem ser mostrados em Informações de exibição.
Verifique a GPU a partir do Gestor de Tarefas
- Clique com o botão direito do rato no ícone do menu Iniciar do Windows e selecione Gestor de Tarefas.
- Selecione Desempenho.
- No último painel da guia, escolha GPU. Se esta opção estiver disponível, provavelmente estará na parte inferior da lista.
- No canto superior direito da seleção de GPU, as informações sobre a GPU do seu computador estarão visíveis.
Não vejo a minha GPU em Definições ou no Gestor de Tarefas, mas sei que tenho uma GPU NVIDIA.
- Abra o Gestor de Dispositivos
- Olhe para adaptadores de vídeo
- Instale o driver apropriado para sua GPU.
Como posso ver a versão do CUDA que tenho?
- Abrir uma janela do PowerShell ou da linha de comando
- Digite
nvcc --version
CUDA não está disponível, confirme que você tem uma GPU suportada pelo CUDA
- Abra o aplicativo GeForce Experience .
- 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/.
- Instale os drivers mais recentes.