Crear una imagen a partir de una instancia de máquina virtual de aprendizaje profundo

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 temas siguientes para crear una instancia. Asegúrate de incluir al menos una GPU en tu nueva instancia.

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:

  1. Conéctate por SSH a tu imagen:
    gcloud compute ssh "DEPLOYMENT_NAME"
  2. 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_NAME

Cuando el comando haya terminado de ejecutarse, 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.