Integrasi PyTorch

Halaman ini menjelaskan integrasi PyTorch Gemini Enterprise Agent Platform dan menyediakan referensi yang menunjukkan cara menggunakan PyTorch di Gemini Enterprise Agent Platform. Integrasi PyTorch Gemini Enterprise Agent Platform memudahkan Anda melatih, men-deploy, dan mengorkestrasi model PyTorch dalam produksi.

Menjalankan kode di notebook

Agent Platform menyediakan dua opsi untuk menjalankan kode Anda di notebook, Colab Enterprise dan Vertex AI Workbench. Untuk mempelajari opsi ini lebih lanjut, lihat memilih solusi notebook.

Container bawaan untuk pelatihan

Gemini Enterprise Agent Platform menyediakan image container Docker bawaan untuk pelatihan model. Container ini diatur menurut framework machine learning dan versi framework serta menyertakan dependensi umum yang mungkin ingin Anda gunakan dalam kode pelatihan. Untuk mempelajari versi PyTorch yang memiliki container pelatihan bawaan dan cara melatih model dengan container pelatihan bawaan, lihat Container bawaan untuk pelatihan kustom.

Container bawaan untuk menayangkan inferensi

Gemini Enterprise Agent Platform menyediakan image container Docker bawaan untuk menayangkan inferensi batch dan online. Container ini diatur menurut framework machine learning dan versi framework serta menyertakan dependensi umum yang mungkin ingin Anda gunakan dalam kode inferensi. Untuk mempelajari versi PyTorch yang memiliki container inferensi bawaan dan cara menayangkan model dengan container inferensi bawaan, lihat Container bawaan untuk pelatihan kustom.

Pelatihan terdistribusi

Anda dapat menjalankan pelatihan terdistribusi model PyTorch di Gemini Enterprise Agent Platform. Untuk pelatihan multi-pekerja, Anda dapat menggunakan Reduction Server untuk mengoptimalkan performa lebih baik lagi untuk operasi kolektif dengan pengurangan penuh. Untuk mempelajari pelatihan terdistribusi di Gemini Enterprise Agent Platform lebih lanjut, lihat Pelatihan terdistribusi.

Referensi untuk menggunakan PyTorch di Gemini Enterprise Agent Platform

Untuk mempelajari lebih lanjut dan mulai menggunakan PyTorch di Gemini Enterprise Agent Platform, lihat referensi berikut:

Langkah berikutnya

  • Tutorial: Menggunakan Gemini Enterprise Agent Platform untuk melatih model klasifikasi gambar PyTorch di salah satu lingkungan container bawaan Gemini Enterprise Agent Platform menggunakan konsol. Google Cloud

    Untuk mengikuti panduan langkah demi langkah untuk tugas ini langsung di Google Cloud konsol, klik Pandu saya:

    Pandu saya