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

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 pra-instal dengan Profiler SageMaker

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

PyTorchgambar

PyTorch versi	AWS DLCgambar URI
2.2.0	763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.2.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.1.0	763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.1.0-gpu-py310-cu121-ubuntu20.04-sagemaker
2.0.1	763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu118-ubuntu20.04-sagemaker 763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:2.0.1-gpu-py310-cu121-ubuntu20.04-sagemaker
1.13.1	763104351884.dkr.ecr.<region>.amazonaws.com/pytorch-training:1.13.1-gpu-py39-cu117-ubuntu20.04-sagemaker

TensorFlow gambar

TensorFlow versi	AWS DLCgambar URI
2.13.0	763104351884.dkr.ecr.<region>.amazonaws.com/tensorflow-training:2.13.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.12.0	763104351884.dkr.ecr.<region>.amazonaws.com/tensorflow-training:2.12.0-gpu-py310-cu118-ubuntu20.04-sagemaker
2.11.0	763104351884.dkr.ecr.<region>.amazonaws.com/tensorflow-training:2.11.0-gpu-py39-cu112-ubuntu20.04-sagemaker

penting

Distribusi dan pemeliharaan container framework pada tabel sebelumnya berada di bawah Framework Support Policy 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 Profiler SageMaker Python yang disediakan di bagian berikut.

SageMaker Profiler Python paket file biner

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

PyTorch

• Python3.8, 11.3: CUDA https://smppy.s3.amazonaws.com/pytorch/cu113/smprof-0.3.334-cp38-cp38-linux_x86_64.whl

• Python3.9, 11.7: CUDA https://smppy.s3.amazonaws.com/pytorch/cu117/smprof-0.3.334-cp39-cp39-linux_x86_64.whl

• Python3.10, 11.8: CUDA https://smppy.s3.amazonaws.com/pytorch/cu118/smprof-0.3.334-cp310-cp310-linux_x86_64.whl

• Python3.10, 12.1: CUDA https://smppy.s3.amazonaws.com/pytorch/cu121/smprof-0.3.334-cp310-cp310-linux_x86_64.whl

TensorFlow

• Python3.9, 11.2: CUDA https://smppy.s3.amazonaws.com/tensorflow/cu112/smprof-0.3.334-cp39-cp39-linux_x86_64.whl

• Python3.10, 11.8: CUDA https://smppy.s3.amazonaws.com/tensorflow/cu118/smprof-0.3.334-cp310-cp310-linux_x86_64.whl

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

Didukung Wilayah AWS

SageMaker Profiler tersedia di berikut Wilayah AWS ini.

• 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 contoh berikut.

CPUdan GPU pembuatan profil

• ml.g4dn.12xlarge

• ml.g5.24xlarge

• ml.g5.48xlarge

• ml.p3dn.24xlarge

• ml.p4de.24xlarge

• ml.p4d.24xlarge

• ml.p5.48xlarge

GPUhanya membuat profil

• ml.g5.2xlarge

• ml.g5.4xlarge

• ml.g5.8xlarge

• ml.g5.16.xlarge