L'installation de pilotes NVIDIA sur une nouvelle instance de VM peut prendre beaucoup de temps, en particulier si vous créez de nombreuses images. Pour éviter cela, vous pouvez créer votre propre image basée sur l'une des instances Deep Learning VM Image, mais sur laquelle les pilotes NVIDIA sont déjà pré-installés.
Cet article explique comment créer une image basée sur une instance Deep Learning VM Image existante.
Créer une instance
Tout d'abord, suivez les instructions de l'un des articles suivants pour créer une instance. Veillez à inclure au moins un GPU dans votre nouvelle instance.
- Créer une instance Deep Learning VM TensorFlow
- Créer une instance Deep Learning VM PyTorch
- Créer une instance Deep Learning VM depuis la ligne de commande
- Créer une instance Deep Learning VM depuis Cloud Marketplace
Vérifier l'installation du pilote NVIDIA
Une fois l'instance démarrée, vérifiez que le pilote NVIDIA a été installé :
- Connectez-vous en SSH à votre image :
gcloud compute ssh "DEPLOYMENT_NAME"
- Exécutez la commande suivante :
nvidia-smi
Si les pilotes ont été installés, vous pouvez continuer.
Arrêter l'instance
Pour arrêter l'instance, exécutez la commande suivante :
gcloud compute instances stop "DEPLOYMENT_NAME"
Créer votre propre image
Créez maintenant votre propre image basée sur l'instance arrêtée. Exécutez la commande suivante sur la ligne de commande, en donnant à la nouvelle image un nom et un nom de famille :
gcloud compute images create "NEW_IMAGE_NAME" \ --source-disk DEPLOYMENT_NAME \ --source-disk-zone ZONE \ --family NEW_FAMILY_NAME
Une fois la commande exécutée, vous disposez d'une nouvelle image dotée de pilotes NVIDIA pré-installés. Vous pouvez l'utiliser pour créer des instances Deep Learning VM.