AWS Deep Learning ARM64 AMI GPU PyTorch 2.4 (Ubuntu 22.04) - AWS Deep Learning AMIs

翻訳は機械翻訳により提供されています。提供された翻訳内容と英語版の間で齟齬、不一致または矛盾がある場合、英語版が優先します。

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

開始方法については、「DLAMI の使用開始」を参照してください。

AMI 名の形式

  • Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.4.${PATCH_VERSION} (Ubuntu 22.04) ${YYYY-MM-DD}

サポートされる EC2 インスタンス

  • G5g

AMI には、以下が含まれます。

  • サポートされているAWSサービス: Amazon EC2

  • オペレーティングシステム: Ubuntu 22.04

  • コンピューティングアーキテクチャ: ARM64

  • Python: /opt/conda/envs/pytorch/bin/python

  • Python バージョン: 3.11

  • Nvidia Driver:

    • OSS Nvidia Driver: 550.144.03

  • NVIDIA CUDA12.1 スタック:

    • CUDA、NCCL、および cuDDN インストールパス: /usr/local/cuda-12.4/

    • デフォルトの CUDA: 12.4

      • /usr/local/cuda のパスは /usr/local/cuda-12.4/ を指します

      • 以下の環境変数が更新されました。

        • /usr/local/cuda/lib:/usr/local/cuda/lib64:/usr/local/cuda:/usr/local/cuda/targets/sbsa-linux/lib:/usr/local/cuda/nvvm/lib64:/usr/local/cuda/extras/CUPTI/lib64 を持つ LD_LIBRARY_PATH

        • パスに /usr/local/cuda/bin/:/usr/local/cuda/include/ を設定しました

    • コンパイル済みシステム NCCL バージョンが /usr/local/cuda/: 2.21.5 に存在します

    • PyTorch conda 環境からの PyTorch のコンパイル済み NCCL のバージョン: 2.20.5

  • aws2 としての AWS CLI v2 と aws としての AWS CLI v1

  • 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.4-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.4.? (Ubuntu 22.04) ????????' 'Name=state,Values=available' \ --query 'reverse(sort_by(Images, &CreationDate))[:1].ImageId' \ --output text

リリース日: 2025 年 2 月 17 日

AMI 名: Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.4.0 (Ubuntu 22.04) 20250215

更新
  • NVIDIA Container Toolkit がバージョン 1.17.3 からバージョン 1.17.4 に更新されました

    • 詳細については、こちら https://github.com/NVIDIA/nvidia-container-toolkit/releases/tag/v1.17.4 のリリースノートページを参照してください。

    • Container Toolkit バージョン 1.17.4 では、CUDA 互換ライブラリのマウントが無効になりました。コンテナワークフローで複数の CUDA バージョンとの互換性を確保するには、「CUDA 互換性レイヤーを使用する場合」のチュートリアルに示されているように、LD_LIBRARY_PATH を更新して CUDA 互換ライブラリを含めるようにしてください。

削除済み

リリース日: 2025 年 1 月 21 日

AMI 名: Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.4.0 (Ubuntu 22.04) 20250117

更新

リリース日: 2024 年 9 月 30 日

AMI 名: Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.4.0 (Ubuntu 22.04) 20240927

更新
  • Nvidia Container Toolkit がバージョン 1.16.1 から 1.16.2 にアップグレードされ、セキュリティの脆弱性 CVE-2024-0133 に対応しました。

リリース日: 2024 年 9 月 26 日

AMI 名: Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.4.0 (Ubuntu 22.04) 20240926

を追加
  • Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.4.0 (Ubuntu 22.04) シリーズの初回リリース。conda 環境を含む PyTorch は、Nvidia Driver R550、CUDA=12.4、cuDNN=8.9.7、PyTorch NCCL=2.20.5 で補完されています。