AWS Pembelajaran Mendalam AMI GPU PyTorch 2.6 (Amazon Linux 2023)

• Pembelajaran Mendalam OSS NVIDIA Driver AMI GPU PyTorch 2.6.0 (Amazon Linux 2023) $ {YYYY-MM-DD}

• Silakan lihat Perubahan penting pada DLAMI

• Pembelajaran Mendalam dengan OSS NVIDIA Driver mendukung G4dn, G5, G6, Gr6, G6e, P4d, P4de, P5, P5e, P5en

• AWS Layanan yang Didukung: EC2

• Sistem Operasi: Amazon Linux 2023

• Arsitektur Komputasi: x86

• NVIDIA CUDA12 .6 tumpukan:

  • Jalur instalasi CUDA, NCCL dan cudDN:/-12.6/usr/local/cuda

  • CUDA standar: 12.6

    • JALAN/usr/local/cuda points to /usr/local/cuda-12.6/

    • Diperbarui di bawah env vars:

      • LD_LIBRARY_PATH memiliki/usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cuda:/usr/local/cud/targets/x86_64-linux/lib

      • PATH untuk memiliki/usr/local/cuda/bin/:/usr/local/cuda/include/

  • Versi NCCL yang dikompilasi untuk 12.6:2.24.3

• Lokasi Tes NCCL:

  • all_reduce, all_gather dan reduce_scatter:/-cuda-xx.x/ usr/local/cuda-xx.x/efa/test

  • Untuk menjalankan pengujian NCCL, LD_LIBRARY_PATH sudah diperbarui dengan jalur yang diperlukan.

    • Umum sudah PATHs ditambahkan ke LD_LIBRARY_PATH:

      • /opt/amazon/efa/lib:/opt/amazon/openmpi/lib:/opt/aws-ofi-nccl/lib:/usr/local/lib:/usr/lib

    • LD_LIBRARY_PATH diperbarui dengan jalur versi CUDA

      • /usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cuda:/usr/local/cud/targets/x86_64-linux/lib

• Pemasang EFA: 1.38.0

• Nvidia GDRCopy: 2.4.1

• AWS OFI NCCL: 1.13.2-aws

  • AWS OFI NCCL sekarang mendukung beberapa versi NCCL dengan build tunggal

  • Jalur instalasi:/ditambahkan ke opt/amazon/ofi-nccl/ . Path /opt/amazon/ofi-nccl/lib LD_LIBRARY_PATH.

• Versi Python: 3.12

• Python:/opt/pytorch/bin/python

• Pengemudi NVIDIA: 570.86.15

• AWS CLI v2 di/usr/bin/aws

• Jenis volume EBS: gp3

• NVMe Lokasi Penyimpanan Instance (pada EC2 instance yang Didukung):/opt/dlami/nvme

• Kueri AMI-ID dengan Parameter SSM (contoh Wilayah adalah us-east-1):

  • Pengemudi OSS Nvidia:

    aws ssm get-parameter --region us-east-1 \
        --name /aws/service/deeplearning/ami/x86_64/oss-nvidia-driver-gpu-pytorch-2.6-amazon-linux-2023/latest/ami-id  \
        --query "Parameter.Value" \
        --output text

• Kueri AMI-ID dengan AWSCLI (contoh Wilayah adalah us-east-1):

  • Pengemudi OSS Nvidia:

    aws ec2 describe-images --region us-east-1 \
        --owners amazon --filters 'Name=name,Values=Deep Learning OSS Nvidia Driver AMI GPU PyTorch 2.6.? (Amazon Linux 2023) ????????' 'Name=state,Values=available' \
        --query 'reverse(sort_by(Images, &CreationDate))[:1].ImageId' \
        --output text

• DeviceIndex unik untuk masing-masing NetworkCard, dan harus berupa bilangan bulat non-negatif kurang dari batas per. ENIs NetworkCard Pada P5, jumlah ENIs per NetworkCard adalah 2, yang berarti bahwa satu-satunya nilai yang valid untuk DeviceIndex adalah 0 atau 1. Di bawah ini adalah contoh perintah peluncuran instance EC2 P5 menggunakan awscli yang menunjukkan NetworkCardIndex dari nomor 0-31 dan DeviceIndex sebagai 0 untuk antarmuka pertama dan DeviceIndex sebagai 1 untuk istirahat 31 antarmuka.

Kernel

• Versi kernel disematkan menggunakan perintah: 

  sudo dnf versionlock kernel*

• Kami menyarankan pengguna untuk menghindari memperbarui versi kernel mereka (kecuali karena patch keamanan) untuk memastikan kompatibilitas dengan driver yang diinstal dan versi paket. Jika pengguna masih ingin memperbarui, mereka dapat menjalankan perintah berikut untuk melepas pin versi kernel mereka: 

  sudo dnf versionlock delete kernel*
  sudo dnf update -y

• Untuk setiap versi baru DLAMI, kernel kompatibel terbaru yang tersedia digunakan.

• Rilis awal Deep Learning OSS Nvidia Driver AMI GPU PyTorch 2.6 untuk Amazon Linux 2023

  • Pada PyTorch 2.6, Pytorch telah menghentikan dukungan untuk Conda. Akibatnya, Pytorch 2.6 dan di atasnya akan beralih menggunakan Lingkungan Virtual Python. Untuk mengaktifkan pytorch venv, gunakan sumber/opt/pytorch/bin/activate