Spesifikasi SageMaker gambar kustom - Amazon SageMaker
Contoh Dockerfile

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

Spesifikasi SageMaker gambar kustom

penting

Pada 30 November 2023, pengalaman Amazon SageMaker Studio sebelumnya sekarang bernama Amazon SageMaker Studio Classic. Bagian berikut khusus untuk menggunakan aplikasi Studio Classic. Untuk informasi tentang menggunakan pengalaman Studio yang diperbarui, lihat SageMaker Studio Amazon.

Spesifikasi berikut berlaku untuk gambar kontainer yang diwakili oleh versi SageMaker gambar.

Menjalankan gambar

ENTRYPOINTdan CMD instruksi diganti untuk mengaktifkan gambar berjalan sebagai aplikasi. KernelGateway

Port 8888 pada gambar dicadangkan untuk menjalankan server KernelGateway web.

Menghentikan gambar

DeleteAppAPIMasalah yang setara dengan docker stop perintah. Proses lain dalam wadah tidak akan mendapatkan SIGTERM sinyalSIGKILL/.

Penemuan kernel

SageMaker mengenali kernel seperti yang didefinisikan oleh spesifikasi kernel Jupyter.

Anda dapat menentukan daftar kernel yang akan ditampilkan sebelum menjalankan gambar. Jika tidak ditentukan, python3 ditampilkan. Gunakan DescribeAppImageConfigAPIuntuk melihat daftar kernel.

Lingkungan Conda diakui sebagai spesifikasi kernel secara default.

Sistem file

/opt/mlDirektori /opt/.sagemakerinternal dan dicadangkan. Data apa pun di direktori ini mungkin tidak terlihat saat runtime.

Data pengguna

Setiap pengguna dalam domain mendapatkan direktori pengguna pada volume Amazon Elastic File System bersama dalam gambar. Lokasi direktori pengguna saat ini pada EFS volume Amazon dapat dikonfigurasi. Secara default, lokasi direktori adalah/home/sagemaker-user.

SageMaker mengkonfigurasi POSIXUID/GIDpemetaan antara gambar dan host. Ini default untuk memetakan pengguna rootUID/GID(0/0) ke/pada UID host. GID

Anda dapat menentukan nilai-nilai ini menggunakan CreateAppImageConfigAPI.

GID/UIDbatas

Amazon SageMaker Studio Classic hanya mendukung yang berikut DefaultUID dan DefaultGID kombinasi:

  • DefaultUID: 1000 dan DefaultGID: 100, yang sesuai dengan pengguna non-priveleged.

  • DefaultUID: 0 dan DefaultGID: 0, yang sesuai dengan akses root.

Metadata

File metadata terletak di. /opt/ml/metadata/resource-metadata.json Tidak ada variabel lingkungan tambahan yang ditambahkan ke variabel yang ditentukan dalam gambar. Untuk informasi selengkapnya, lihat Dapatkan Metadata Aplikasi.

GPU

Pada sebuah GPU contoh, gambar dijalankan dengan --gpus opsi. Hanya CUDA toolkit yang harus disertakan dalam gambar bukan NVIDIA driver. Untuk informasi selengkapnya, lihat Panduan NVIDIA Pengguna.

Metrik dan pencatatan

Log dari KernelGateway proses dikirim ke Amazon CloudWatch di akun pelanggan. Nama grup log adalah/aws/sagemaker/studio. Nama aliran log adalah$domainID/$userProfileName/KernelGateway/$appName.

Ukuran gambar

Terbatas hingga 35 GB. Untuk melihat ukuran gambar Anda, jalankandocker image ls.

Contoh Dockerfile

Contoh berikut Dockerfile membuat Amazon Linux 2 berbasis gambar, menginstal paket pihak ketiga dan python3 kernel, dan menetapkan cakupan ke pengguna yang tidak memiliki hak istimewa.

FROM public.ecr.aws/amazonlinux/amazonlinux:2

ARG NB_USER="sagemaker-user"
ARG NB_UID="1000"
ARG NB_GID="100"

RUN \
    yum install --assumeyes python3 shadow-utils && \
    useradd --create-home --shell /bin/bash --gid "${NB_GID}" --uid ${NB_UID} ${NB_USER} && \
    yum clean all && \
    python3 -m pip install ipykernel && \
    python3 -m ipykernel install

USER ${NB_UID}