AWS GPU ARM64 AMI de aprendizado profundo PyTorch 2.5 (Ubuntu 22.04) - AMIs de deep learning da AWS

As traduções são geradas por tradução automática. Em caso de conflito entre o conteúdo da tradução e da versão original em inglês, a versão em inglês prevalecerá.

AWS GPU ARM64 AMI de aprendizado profundo PyTorch 2.5 (Ubuntu 22.04)

Para obter ajuda para começar, consulteConceitos básicos da DLAMI.

Formato de nome da AMI

  • ARM64 AMI OSS Nvidia Driver PyTorch GPU 2.5 de aprendizado profundo. $ {PATCH-VERSION} (Ubuntu 22.04) $ {YYYY-MM-DD}

EC2 Instâncias suportadas

  • G5g

A AMI inclui o seguinte:

  • AWS Serviço suportado: Amazon EC2

  • Sistema operacional: Ubuntu 22.04

  • Arquitetura de computação: ARM64

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

  • Versão Python: 3.11

  • Controlador NVIDIA:

    • Driver OSS Nvidia: 550.144.03

  • Pilha NVIDIA CUDA12 4.4:

    • Caminho de instalação de CUDA, NCCL e cuDDN:/-12.4/ usr/local/cuda

    • CUDA padrão: 12,4

      • CAMINHO/usr/local/cuda points to /usr/local/cuda-12.4/

      • Variáveis de ambiente atualizadas abaixo:

        • LD_LIBRARY_PATH deve ter/64 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/lib

        • PATH para ter/usr/local/cuda/bin/:/usr/local/cuda/include/

    • Versão NCCL do sistema compilado presente em/usr/local/cuda/: 2.21.5

    • PyTorch Versão NCCL compilada do ambiente PyTorch conda: 2.21.5

  • AWS CLI v2 como aws2 e AWS CLI v1 como aws

  • Tipo de volume do EBS: gp3

  • Consulte o AMI-ID com o parâmetro SSM (a região do exemplo é us-east-1):

    aws ssm get-parameter --region us-east-1 \ --name /aws/service/deeplearning/ami/arm64/oss-nvidia-driver-gpu-pytorch-2.5-ubuntu-22.04/latest/ami-id \ --query "Parameter.Value" \ --output text
  • Consulte o AMI-ID com AWSCLI (a região de exemplo é 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.5.? (Ubuntu 22.04) ????????' 'Name=state,Values=available' \ --query 'reverse(sort_by(Images, &CreationDate))[:1].ImageId' \ --output text

Data de lançamento: 2025-02-17

Nome da AMI: Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.5.1 (Ubuntu 22.04) 20250215

Atualizado

  • Kit de ferramentas de contêiner NVIDIA atualizado da versão 1.17.3 para a versão 1.17.4

    • Consulte a página de notas de lançamento aqui para obter mais informações: https://github.com/NVIDIA/nvidia-container-toolkit/releases/tag/v1.17.4

    • No Container Toolkit versão 1.17.4, a montagem das bibliotecas de compatibilidade CUDA agora está desativada. Para garantir a compatibilidade com várias versões do CUDA em fluxos de trabalho de contêineres, certifique-se de atualizar seu LD_LIBRARY_PATH para incluir suas bibliotecas de compatibilidade CUDA, conforme mostrado no tutorial Se você usar uma camada de compatibilidade CUDA.

Removido

Data de lançamento: 2025-01-21

Nome da AMI: Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.5.1 (Ubuntu 22.04) 20250117

Atualizado

Data de lançamento: 2024-11-22

Nome da AMI: Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.5.1 (Ubuntu 22.04) 20241122

Adicionado

  • Lançamento inicial da série Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.5.1 (Ubuntu 22.04). Incluindo um ambiente conda pytorch complementado com o NVIDIA Driver R550, CUDA=12,4, cuDNN=8.9.7, NCCL=2.21.5. PyTorch