Integrazione di TensorFlow

Questa pagina spiega l'integrazione di TensorFlow di Gemini Enterprise Agent Platform e fornisce risorse che mostrano come utilizzare TensorFlow su Gemini Enterprise Agent Platform. L'integrazione di TensorFlow di Gemini Enterprise Agent Platform semplifica l'addestramento, il deployment e l'orchestrazione dei modelli TensorFlow in produzione.

Eseguire il codice nei notebook

Agent Platform offre due opzioni per l'esecuzione del codice nei notebook: Colab Enterprise e Gemini Enterprise Agent Platform Workbench. Per saperne di più su queste opzioni, consulta Scegliere una soluzione di notebook.

Container predefiniti per l'addestramento

Gemini Enterprise Agent Platform 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.

Per scoprire quali versioni di TensorFlow hanno container di addestramento predefiniti e come addestrare i modelli con un container di addestramento predefinito, consulta Container predefiniti per l'addestramento personalizzato.

Addestramento distribuito

Puoi eseguire l'addestramento distribuito dei modelli TensorFlow su Gemini Enterprise Agent Platform. Per l'addestramento multi-worker, puoi utilizzare il server di riduzione per ottimizzare ulteriormente il rendimento per le operazioni collettive all-reduce. Per saperne di più sull' addestramento distribuito su Gemini Enterprise Agent Platform, consulta Addestramento distribuito.

Container predefiniti per l'inferenza

Analogamente ai container predefiniti per l'addestramento, Gemini Enterprise Agent Platform fornisce immagini di container predefiniti per la pubblicazione di inferenze e spiegazioni da modelli TensorFlow creati all'interno o all'esterno di Gemini Enterprise Agent Platform. Queste immagini forniscono server di inferenza HTTP che puoi utilizzare per pubblicare inferenze con una configurazione minima.

Per scoprire quali versioni di TensorFlow hanno container di addestramento predefiniti e come addestrare i modelli con un container di addestramento predefinito, consulta Container predefiniti per l'addestramento personalizzato.

Runtime TensorFlow ottimizzato

Il runtime TensorFlow ottimizzato utilizza le ottimizzazioni dei modelli e le nuove tecnologie proprietarie di Google per migliorare la velocità e ridurre il costo delle inferenze rispetto ai container di inferenza predefiniti standard di Gemini Enterprise Agent Platform per TensorFlow.

Integrazione di TensorFlow Cloud Profiler

Addestra i modelli in modo più economico e veloce monitorando e ottimizzando il rendimento del job di addestramento utilizzando l'integrazione di TensorFlow Cloud Profiler di Gemini Enterprise Agent Platform. TensorFlow Cloud Profiler ti aiuta a comprendere il consumo di risorse delle operazioni di addestramento in modo da poter identificare ed eliminare i colli di bottiglia delle prestazioni.

Per saperne di più su Agent Platform TensorFlow Cloud Profiler, consulta Profilare il rendimento dell'addestramento dei modelli utilizzando Profiler.

Risorse per l'utilizzo di TensorFlow su Gemini Enterprise Agent Platform

Per saperne di più e iniziare a utilizzare TensorFlow in Gemini Enterprise Agent Platform, consulta le seguenti risorse.