Instalar controladores de NVIDIA en una instancia de VM nueva puede llevar mucho tiempo, sobre todo si creas muchas imágenes. Una forma de evitarlo es crear tu propia imagen basada en una de las imágenes de máquina virtual de aprendizaje profundo, pero que ya tenga los controladores de NVIDIA preinstalados.
En este tema se describe cómo crear una imagen a partir de una imagen de máquina virtual de aprendizaje profundo.
Crear una instancia
Primero, sigue las instrucciones de uno de los siguientes temas para crear una instancia. Asegúrate de incluir al menos una GPU en tu nueva instancia.
- Crear una instancia de VM de aprendizaje profundo de TensorFlow
- Crear una instancia de VM de aprendizaje profundo de PyTorch
- Crear una instancia de VM de aprendizaje profundo desde la línea de comandos
- Crear una instancia de VM de aprendizaje profundo desde Cloud Marketplace
Verificar la instalación del controlador de NVIDIA
Una vez que se haya iniciado la instancia, comprueba que se ha instalado el controlador de NVIDIA:
- Conéctate por SSH a tu imagen:
gcloud compute ssh "DEPLOYMENT_NAME"
- Ejecuta el siguiente comando:
nvidia-smi
Si los controladores se han instalado, puedes continuar.
Detener la instancia
Detén la instancia ejecutando el siguiente comando:
gcloud compute instances stop "DEPLOYMENT_NAME"
Crea tu propia imagen
Ahora puedes crear tu propia imagen basada en la instancia detenida. Ejecuta lo siguiente en la línea de comandos, asignando un nombre y un nombre de familia a la nueva imagen:
gcloud compute images create "NEW_IMAGE_NAME" \
--source-disk DEPLOYMENT_NAME \
--source-disk-zone ZONE \
--family NEW_FAMILY_NAMECuando se haya terminado de ejecutar el comando, tendrás una nueva imagen con los controladores de NVIDIA preinstalados que podrás usar para crear nuevas instancias de máquina virtual de aprendizaje profundo.