L'installazione dei driver NVIDIA su una nuova istanza VM può richiedere molto tempo, soprattutto se stai creando numerose immagini. Un modo per prevenire questo problema è creare un'immagine personalizzata basata su una delle immagini Deep Learning VM, ma che ha già i driver NVIDIA preinstallati.
Questo argomento descrive come creare una nuova immagine basata su un'immagine Deep Learning VM esistente.
Creazione di una nuova istanza
Innanzitutto, segui le istruzioni in uno dei seguenti argomenti per creare una nuova istanza. Assicurati di includere almeno una GPU nella nuova istanza.
- Creazione di un'istanza TensorFlow Deep Learning VM
- Creazione di un'istanza PyTorch Deep Learning VM
- Creazione di un'istanza Deep Learning VM dalla riga di comando
- Creazione di un'istanza Deep Learning VM da Cloud Marketplace
Verifica l'installazione del driver NVIDIA
Dopo l'avvio dell'istanza, verifica che il driver NVIDIA sia stato installato:
- Accedi alla tua immagine tramite SSH:
gcloud compute ssh "DEPLOYMENT_NAME"
- Esegui questo comando:
nvidia-smi
Se i driver sono stati installati, puoi continuare.
Arresta l'istanza
Esegui il comando seguente per arrestare l'istanza:
gcloud compute instances stop "DEPLOYMENT_NAME"
Crea la tua immagine
Ora puoi creare la tua immagine basata sull'istanza arrestata. Esegui il comando seguente nella riga di comando, assegnando alla nuova immagine un nome e una famiglia:
gcloud compute images create "NEW_IMAGE_NAME" \
--source-disk DEPLOYMENT_NAME \
--source-disk-zone ZONE \
--family NEW_FAMILY_NAME
Una volta completata l'esecuzione del comando, avrai una nuova immagine con driver NVIDIA preinstallati che puoi utilizzare per creare nuove istanze Deep Learning VM.