Integrasi PyTorch

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

Menjalankan kode di notebook

Platform Agen 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

Platform Agen Gemini Enterprise 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

Platform Agen Gemini Enterprise 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 Platform Agen Gemini Enterprise. 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 Platform Agen Gemini Enterprise lebih lanjut, lihat Pelatihan terdistribusi.

Referensi untuk menggunakan PyTorch di Platform Agen Gemini Enterprise

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

Langkah berikutnya

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

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

    Pandu saya