Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.

Gambar SageMaker AI Docker bawaan untuk pembelajaran mendalam

Amazon SageMaker AI menyediakan gambar Docker bawaan yang mencakup kerangka kerja pembelajaran mendalam dan dependensi lain yang diperlukan untuk pelatihan dan inferensi. Untuk daftar lengkap gambar Docker bawaan yang dikelola oleh SageMaker AI, lihat Docker Registry Paths dan Contoh Kode.

Menggunakan SageMaker AI Python SDK

Dengan SageMaker Python SDK, Anda dapat melatih dan menerapkan model menggunakan kerangka kerja pembelajaran mendalam yang populer ini. Untuk petunjuk tentang menginstal dan menggunakan SDK, lihat Amazon SageMaker Python SDK . Tabel berikut mencantumkan kerangka kerja yang tersedia dan petunjuk tentang cara menggunakannya dengan SageMaker Python SDK:

Memperluas Gambar Docker SageMaker AI Prebuilt

Anda dapat menyesuaikan wadah bawaan ini atau memperluasnya sesuai kebutuhan. Dengan penyesuaian ini, Anda dapat menangani persyaratan fungsional tambahan untuk algoritma atau model Anda yang tidak didukung oleh gambar SageMaker AI Docker bawaan. Untuk contoh ini, lihat Fine-tuning dan menerapkan model BerTopic pada SageMaker AI dengan skrip dan dataset Anda sendiri, dengan memperluas wadah yang ada PyTorch .

Anda juga dapat menggunakan wadah bawaan untuk menerapkan model kustom atau model yang telah dilatih dalam kerangka kerja selain SageMaker AI. Untuk ikhtisar prosesnya, lihat Mem bawa MXnet atau TensorFlow Model Pra-Training Anda Sendiri ke Amazon. SageMaker Tutorial ini mencakup membawa artefak model terlatih ke dalam SageMaker AI dan menampungnya di titik akhir.