View a markdown version of this page

Gambar kerangka kerja yang didukung, Wilayah AWS, dan jenis instance - Amazon SageMaker AI
SageMaker Gambar kerangka AI yang sudah diinstal sebelumnya dengan Profiler SageMakerSageMaker File biner paket Profiler PythonDidukung Wilayah AWSTipe instans yang didukung

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

Gambar kerangka kerja yang didukung, Wilayah AWS, dan jenis instance

catatan

Pemberitahuan akhir dukungan: Pada 30 Juni 2027, dukungan untuk Amazon SageMaker Profiler AWS akan berakhir. Setelah 30 Juni 2027, Anda tidak akan lagi dapat mengakses konsol Profiler atau sumber daya Profiler. Untuk informasi selengkapnya, lihat Perubahan ketersediaan profiler.

Fitur ini mendukung kerangka kerja pembelajaran mesin berikut dan Wilayah AWS.

catatan

Untuk menggunakan fitur ini, pastikan Anda telah menginstal SageMaker Python SDK versi 2.180.0 atau yang lebih baru.

SageMaker Gambar kerangka AI yang sudah diinstal sebelumnya dengan Profiler SageMaker

SageMaker Profiler sudah diinstal sebelumnya di AWS Deep Learning Containers untuk SageMaker AI berikut.

PyTorch gambar

PyTorch versi AWS URI gambar DLC
2.2.0

763104351884.dkr.cr. <region>.amazonaw. com/pytorch-pelatihan: 2.2.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.1.0

763104351884.dkr.cr. <region>.amazonaw. com/pytorch-pelatihan: 2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.0.1

763104351884.dkr.cr. <region>.amazonaw. com/pytorch-pelatihan: 2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker

763104351884.dkr.cr. <region>.amazonaw. com/pytorch-pelatihan: 2.0.1-gpu-py310-cu121-ubuntu20.04-sagemaker
1.13.1

763104351884.dkr.cr. <region>.amazonaw. com/pytorch-pelatihan: 1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker

TensorFlow gambar

TensorFlow versi AWS URI gambar DLC
2.13.0

763104351884.dkr.cr. <region>.amazonaw. com/tensorflow-pelatihan: 2.13.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.12.0

763104351884.dkr.cr. <region>.amazonaw. com/tensorflow-pelatihan: 2.12.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.11.0

763104351884.dkr.cr. <region>.amazonaw. com/tensorflow-pelatihan: 2.11.0-gpu-py39-cu112-ubuntu20.04-sagemaker
penting

Distribusi dan pemeliharaan wadah kerangka kerja dalam tabel sebelumnya berada di bawah Kebijakan Dukungan Ker angka yang dikelola oleh layanan AWS Deep Learning Containers. Kami sangat menyarankan Anda untuk meningkatkan ke versi kerangka kerja yang saat ini didukung, jika Anda menggunakan versi kerangka kerja sebelumnya yang tidak lagi didukung.

catatan

Jika Anda ingin menggunakan SageMaker Profiler untuk gambar kerangka kerja lain atau gambar Docker Anda sendiri, Anda dapat menginstal SageMaker Profiler menggunakan file biner paket SageMaker Profiler Python yang disediakan di bagian berikut.

SageMaker File biner paket Profiler Python

Jika Anda ingin mengonfigurasi wadah Docker Anda sendiri, gunakan SageMaker Profiler di wadah pra-bangun lainnya untuk PyTorch dan TensorFlow, atau instal paket SageMaker Profiler Python secara lokal, gunakan salah satu file biner berikut. Tergantung pada versi Python dan CUDA di lingkungan Anda, pilih salah satu dari yang berikut ini.

PyTorch

TensorFlow

Untuk informasi selengkapnya tentang cara menginstal SageMaker Profiler menggunakan file biner, lihat(Opsional) Instal SageMaker paket Profiler Python.

Didukung Wilayah AWS

SageMaker Profiler tersedia sebagai berikut Wilayah AWS.

  • US East (N. Virginia) (us-east-1)

  • US East (Ohio) (us-east-2)

  • US West (Oregon) (us-west-2)

  • Europe (Frankfurt) (eu-central-1)

  • Europe (Ireland) (eu-west-1)

Tipe instans yang didukung

SageMaker Profiler mendukung pembuatan profil pekerjaan pelatihan pada jenis instance berikut.

Profil CPU dan GPU

  • ml.g4dn.12xlarge

  • ml.g5.24xlarge

  • ml.g5.48xlarge

  • ml.p3dn.24xlarge

  • ml.p4de.24xlarge

  • ml.p4d.24xlarge

  • ml.p5.48xlarge

Hanya pembuatan profil GPU

  • ml.g5.2xlarge

  • ml.g5.4xlarge

  • ml.g5.8xlarge

  • ml.g5.16.xlarge