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/cuda/targets/sbsa-linux usr/local/cuda/lib:/usr/local/cuda/nvvm/lib64:/usr/local/cuda/extras/CUPTI/lib64 を含める

        • /usr/local/cuda/bin/:/usr/local/cuda/include/ を持つ PATH

    • コンパイル済みシステム NCCL バージョン /usr/local/cuda/: 2.24.3

    • 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 をアクティブ化するには、ソース /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 Driver R570、CUDA=12.6、cuDNN=9.7、PyTorch NCCL=2.21.5 で補完された Python 仮想環境 pytorch (ソース /opt/pytorch/bin/activate) が含まれます。