Instalar controladores NVIDIA en una instancia de VM nueva puede llevar mucho tiempo, en especial si creas varias imágenes. Una forma de ahorrar tiempo es crear tu propia imagen que se base en una de las imágenes de VM de aprendizaje profundo, pero que ya tenga preinstalados los controladores NVIDIA.
En este tema, se describe cómo crear una imagen nueva basada en una imagen existente de VM de aprendizaje profundo.
Crea una instancia nueva
Primero, sigue las instrucciones de uno de los temas siguientes para crear una instancia nueva. Asegúrate de incluir, al menos, una GPU en tu instancia nueva.
- Crea una instancia de VM de aprendizaje profundo de TensorFlow
- Crea una instancia de VM de aprendizaje profundo de PyTorch
- Crea una instancia de VM de aprendizaje profundo desde la línea de comandos
- Crea una instancia de VM de aprendizaje profundo desde Cloud Marketplace
Verifica la instalación del controlador NVIDIA
Una vez que se inicia la instancia, verifica que se haya instalado el controlador NVIDIA:
- Usa SSH para tu imagen:
gcloud compute ssh "DEPLOYMENT_NAME"
- Ejecuta el siguiente comando:
nvidia-smi
Si se instalaron los controladores, puedes continuar.
Detén la instancia
Detén la instancia mediante la ejecución del siguiente comando:
gcloud compute instances stop "DEPLOYMENT_NAME"
Crea tu propia imagen
Ahora, debes crear tu propia imagen basada en la instancia detenida. Ejecuta el siguiente comando en la línea de comandos y asígnale un nombre y un nombre de familia a la imagen nueva:
gcloud compute images create "NEW_IMAGE_NAME" \ --source-disk DEPLOYMENT_NAME \ --source-disk-zone ZONE \ --family NEW_FAMILY_NAME
Cuando el comando termine de ejecutarse, tendrás una imagen nueva con los controladores NVIDIA preinstalados que puedes usar para crear nuevas instancias de VM de aprendizaje profundo.