A instalação de drivers NVIDIA em uma nova instância de VM pode demorar, especialmente se você estiver criando várias imagens. Uma maneira de evitar isso é criar sua própria imagem baseada em uma das imagens de VM de aprendizado profundo, mas que já tenha os drivers NVIDIA pré-instalados.
Neste tópico, descrevemos como criar uma nova imagem com base em uma imagem atual de VM de aprendizado profundo.
Criar uma nova instância
Primeiro, siga as instruções em um dos tópicos a seguir para criar uma nova instância. Não se esqueça de incluir pelo menos uma GPU em sua nova instância.
- Como criar uma instância de VM de aprendizado profundo do TensorFlow
- Como criar uma instância de VM de aprendizado profundo do PyTorch
- Como criar uma instância de VM de aprendizado profundo da linha de comando
- Como criar uma instância de VM de aprendizado profundo do Cloud Marketplace
Verificar a instalação do driver NVIDIA
Após a inicialização da instância, verifique se o driver NVIDIA foi instalado:
- SSH para sua imagem:
gcloud compute ssh "DEPLOYMENT_NAME"
- Execute este comando:
nvidia-smi
Se os drivers foram instalados, continue.
Interromper a instância
Pare a instância executando o comando a seguir:
gcloud compute instances stop "DEPLOYMENT_NAME"
Criar sua própria imagem
Agora você cria sua própria imagem com base na instância parada. Execute o seguinte na linha de comando, fornecendo à nova imagem um nome e um nome de família:
gcloud compute images create "NEW_IMAGE_NAME" \
--source-disk DEPLOYMENT_NAME \
--source-disk-zone ZONE \
--family NEW_FAMILY_NAME
Após a conclusão da execução do comando, você terá uma nova imagem com os drivers NVIDIA pré-instalados que poderão ser usados para criar novas instâncias de VM de aprendizado profundo.