A instalação de controladores NVIDIA numa nova instância de VM pode demorar algum tempo, especialmente se estiver a criar várias imagens. Uma forma de evitar esta situação é criar a sua própria imagem baseada numa das imagens de VM de aprendizagem profunda, mas que já tenha os controladores NVIDIA pré-instalados.
Este tópico descreve como criar uma nova imagem com base numa imagem de VM de aprendizagem profunda existente.
Crie uma nova instância
Primeiro, siga as instruções num dos seguintes tópicos para criar uma nova instância. Não se esqueça de incluir, pelo menos, uma GPU na nova instância.
- Criar uma instância de VM de aprendizagem avançada do TensorFlow
- Criar uma instância de VM de aprendizagem avançada do PyTorch
- Criar uma instância de VM de aprendizagem avançada a partir da linha de comandos
- Criar uma instância de VM de aprendizagem avançada a partir do Cloud Marketplace
Valide a instalação do controlador NVIDIA
Assim que a instância for iniciada, verifique se o controlador NVIDIA foi instalado:
- SSH para a sua imagem:
gcloud compute ssh "DEPLOYMENT_NAME"
- Execute o seguinte comando:
nvidia-smi
Se os controladores tiverem sido instalados, pode continuar.
Pare a instância
Pare a instância executando o seguinte comando:
gcloud compute instances stop "DEPLOYMENT_NAME"
Crie a sua própria imagem
Agora, pode criar a sua própria imagem com base na instância parada. Execute o seguinte comando na linha de comandos, atribuindo um nome e um nome de família à nova imagem:
gcloud compute images create "NEW_IMAGE_NAME" \ --source-disk DEPLOYMENT_NAME \ --source-disk-zone ZONE \ --family NEW_FAMILY_NAME
Quando o comando terminar de ser executado, tem uma nova imagem com os controladores NVIDIA pré-instalados que pode usar para criar novas instâncias de VMs de aprendizagem profunda.