L'installazione dei driver NVIDIA su una nuova istanza VM richiedere molto tempo, soprattutto se ne devi creare numerose. Solo andata evitare che ciò accada è creare una tua immagine basata su uno dei Deep Learning VM Image, ma che dispongono già dei driver NVIDIA sono preinstallate.
Questo argomento descrive come creare una nuova immagine basata su un modello Deep Learning VM Image.
Crea una nuova istanza
Innanzitutto, segui le istruzioni in uno dei seguenti argomenti per creare un nuovo in esecuzione in un'istanza Compute Engine. Assicurati di includere almeno una GPU nella nuova istanza.
- Creazione di un'istanza VM di deep learning di TensorFlow
- Creazione di un'istanza VM di deep learning PyTorch
- Creazione di un'istanza VM di deep learning dal comando Riga
- Creazione di un'istanza VM di deep learning da Cloud Marketplace
Verificare l'installazione del driver NVIDIA
Dopo l'avvio dell'istanza, verifica che il driver NVIDIA sia stato installato:
- Accedi all'immagine tramite SSH:
gcloud compute ssh "DEPLOYMENT_NAME"
- Esegui questo comando:
nvidia-smi
Se i driver sono stati installati, puoi continuare.
Arresta l'istanza
Arresta l'istanza eseguendo questo comando:
gcloud compute instances stop "DEPLOYMENT_NAME"
Crea la tua immagine
Ora crei la tua immagine basata sull'istanza arrestata. Esegui questo comando nella riga di comando, assegnando alla nuova immagine un nome e un cognome:
gcloud compute images create "NEW_IMAGE_NAME" \
--source-disk DEPLOYMENT_NAME \
--source-disk-zone ZONE \
--family NEW_FAMILY_NAMEAl termine dell'esecuzione del comando, avrai una nuova immagine con i driver NVIDIA preinstallati che puoi utilizzare per creare nuove istanze Deep Learning VM.