Terjemahan disediakan oleh mesin penerjemah. Jika konten terjemahan yang diberikan bertentangan dengan versi bahasa Inggris aslinya, utamakan versi bahasa Inggris.
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
SageMaker Gambar kerangka kerja AI yang sudah diinstal sebelumnya dengan Profiler SageMaker
SageMaker Profiler sudah diinstal sebelumnya di AWS Deep Learning Containers for SageMaker AI
PyTorchgambar
PyTorch versi | AWS URI gambar DLC |
---|---|
2.2.0 |
|
2.1.0 |
|
2.0.1 |
|
1.13.1 |
|
TensorFlow gambar
TensorFlow versi | AWS URI gambar DLC |
---|---|
2.13.0 |
|
2.12.0 |
|
2.11.0 |
|
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
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. Bergantung pada versi Python dan CUDA di lingkungan Anda, pilih salah satu dari yang berikut ini.
PyTorch
-
Python3.8, CUDA 11.3:
https://smppy.s3.amazonaws.com/pytorch/cu113/smprof-0.3.334-cp38-cp38-linux_x86_64.whl
-
Python3.9, CUDA 11.7:
https://smppy.s3.amazonaws.com/pytorch/cu117/smprof-0.3.334-cp39-cp39-linux_x86_64.whl
-
Python3.10, CUDA 11.8:
https://smppy.s3.amazonaws.com/pytorch/cu118/smprof-0.3.334-cp310-cp310-linux_x86_64.whl
-
Python3.10, CUDA 12.1:
https://smppy.s3.amazonaws.com/pytorch/cu121/smprof-0.3.334-cp310-cp310-linux_x86_64.whl
TensorFlow
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.
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 profil GPU
-
ml.g5.2xlarge
-
ml.g5.4xlarge
-
ml.g5.8xlarge
-
ml.g5.16.xlarge