Integrazione di PyTorch

Questa pagina spiega l'integrazione di PyTorch di Vertex AI e fornisce risorse che mostrano come utilizzare PyTorch su Vertex AI. PyTorch di Vertex AI semplifica l'addestramento, il deployment e l'orchestrazione di PyTorch modelli in produzione.

Eseguire codice nei blocchi note

Vertex AI offre due opzioni per eseguire il codice nei notebook: Colab Enterprise e Vertex AI Workbench. Per scoprire di più su queste opzioni, vedi scegli una soluzione di blocco note.

Container predefiniti per l'addestramento

Vertex AI fornisce immagini di container Docker predefiniti per l'addestramento dei modelli. Questi container sono organizzati per framework di machine learning e versioni del framework e includono dipendenze comuni che potresti voler utilizzare nel codice di addestramento. Informazioni sulle versioni di PyTorch con addestramento predefinito di addestramento e come addestrare i modelli con un container di addestramento predefinito, Container predefiniti per l'addestramento personalizzato.

Container predefiniti per la fornitura di previsioni

Vertex AI fornisce immagini di container Docker predefiniti per fornire sia le previsioni in batch sia quelle online. Questi container sono organizzati per framework di machine learning e versioni del framework e includono dipendenze comuni che potresti voler utilizzare nel codice di previsione. Per scoprire quali versioni di PyTorch dispongono di container di previsione predefiniti e come pubblicare modelli con un container di previsione predefinito, consulta Container predefiniti per l'addestramento personalizzato.

Addestramento distribuito

Puoi eseguire l'addestramento distribuito dei modelli PyTorch su Vertex AI. Per con più lavoratori, puoi utilizzare Reduct Server per ottimizzare le prestazioni ulteriormente per ridurre del tutto le operazioni collettive. Per scoprire di più su per l'addestramento distribuito su Vertex AI, Addestramento distribuito.

Risorse per l'utilizzo di PyTorch su Vertex AI

Per scoprire di più e iniziare a utilizzare PyTorch in Vertex AI, consulta le seguenti risorse:

Passaggi successivi

  • Tutorial: utilizza Vertex AI per addestrare un modello di classificazione delle immagini PyTorch in uno degli ambienti container predefiniti di Vertex AI utilizzando la console Google Cloud.

    Per seguire le indicazioni dettagliate per questa attività direttamente nella console Google Cloud, fai clic su Procedura guidata:

    Procedura guidata