AWS Deep Learning ARM64 AMI GPU PyTorch 2.6(Ubuntu 22.04) - AWS Deep Learning AMIs
AMI 이름 형식지원되는 EC2 인스턴스AMI 콘텐츠고지 사항릴리스 날짜: 2025-02-21

기계 번역으로 제공되는 번역입니다. 제공된 번역과 원본 영어의 내용이 상충하는 경우에는 영어 버전이 우선합니다.

AWS Deep Learning ARM64 AMI GPU PyTorch 2.6(Ubuntu 22.04)

시작하는 데 도움이 필요하면 섹션을 참조하세요DLAMI 시작하기.

AMI 이름 형식

  • Deep Learning ARM64 AMI OSS NVIDIA 드라이버 GPU PyTorch 2.6.${PATCH-VERSION}(Ubuntu 22.04) ${YYYY-MM-DD}

지원되는 EC2 인스턴스

  • G5g

AMI에는 다음이 포함됩니다.

  • 지원되는 AWS 서비스: Amazon EC2

  • 운영 체제: Ubuntu 22.04

  • 컴퓨팅 아키텍처: ARM64

  • Python: /opt/pytorch/bin/python

  • Python 버전:3.12

  • NVIDIA 드라이버:

    • OSS Nvidia 드라이버:570.86.15

  • NVIDIA CUDA12.6 스택:

    • CUDA, NCCL 및 cuDDN 설치 경로: /usr/local/cuda-12.6/

    • 기본 CUDA:12.6

      • PATH /usr/local/cudapoints에서 /usr/local/cuda-12.6/로

      • 아래 env vars가 업데이트되었습니다.

        • LD_LIBRARY_PATH: /usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cudausr/local/cudatargets/sbsa-linux/lib:/usr/local/cuda/nvm/lib64:/usr/local/cuda/extras/CUPTI/lib64

        • PATH에 /usr/local/cuda/bin/:/usr/local/cuda/include/

    • /usr/local/cuda/: 2.24.3에 있는 컴파일된 시스템 NCCL 버전

    • PyTorch venv 환경의 PyTorch 컴파일 NCCL 버전: 2.21.5

  • AWS CLI v2as aws2 및AWS CLI v1as aws

  • EBS 볼륨 유형: gp3

  • SSM 파라미터를 사용하여 AMI-ID 쿼리(예제 리전은 us-east-1):

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

  • AWSCLI를 사용하여 AMI-ID 쿼리(예제 리전은 us-east-1):

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

고지 사항

Anaconda 채널의 PyTorch 사용 중단

PyTorch 2.6부터 PyTorch는 Conda에 대한 지원을 중단했습니다(공식 발표 참조). 따라서 PyTorch 2.6 이상은 Python 가상 환경을 사용하여 로 이동합니다. PyTorch venv를 활성화하려면 source /opt/pytorch/bin/activate를 사용하세요. 

릴리스 날짜: 2025-02-21

AMI 이름: Deep Learning ARM64 AMI OSS NVIDIA 드라이버 GPU PyTorch 2.6.0(Ubuntu 22.04) 20250221

추가됨

  • Deep Learning ARM64 AMI OSS Nvidia 드라이버 GPU PyTorch 2.6.0(Ubuntu 22.04) 시리즈의 최초 릴리스입니다. NVIDIA 드라이버 R570, CUDA=12.6, cuDNN=9.7, PyTorch NCCL=2.21.5로 보완된 Python 가상 환경 pytorch(소스 /opt/pytorch/bin/activate)를 포함합니다.