L'installazione dei driver NVIDIA su una nuova istanza VM può essere laboriosa, soprattutto se crei numerose immagini. Un modo per evitarlo è creare un'immagine basata su una delle immagini VM di deep learning, ma con i driver NVIDIA già preinstallati.
Questo argomento descrive come creare una nuova immagine basata su un'immagine VM di deep learning esistente.
Crea 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 VM di deep learning di TensorFlow
- Creazione di un'istanza VM di deep learning PyTorch
- Creazione di un'istanza VM di deep learning dalla riga di comando
- Creazione di un'istanza VM di deep learning da Cloud Marketplace
Verificare l'installazione del driver NVIDIA
Una volta avviata l'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
Interrompi l'istanza eseguendo il seguente comando:
gcloud compute instances stop "DEPLOYMENT_NAME"
Creare un'immagine
Ora crea la tua immagine in base all'istanza interrotta. Esegui quanto segue nella riga di comando, assegnando alla nuova immagine un nome e un nome di famiglia:
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.