From f81f48f44d6f243bd86f1ce2240ed1ed6b4a8a33 Mon Sep 17 00:00:00 2001 From: Maxwell Nuyens Date: Fri, 22 Nov 2024 20:24:36 -0800 Subject: [PATCH] fix: Updating Dockerfile to pin versions and fix the exampel (#505) --- 3.test_cases/10.FSDP/Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/3.test_cases/10.FSDP/Dockerfile b/3.test_cases/10.FSDP/Dockerfile index 2ad6af05..0d7f2db0 100644 --- a/3.test_cases/10.FSDP/Dockerfile +++ b/3.test_cases/10.FSDP/Dockerfile @@ -3,8 +3,8 @@ FROM public.ecr.aws/hpc-cloud/nccl-tests:latest RUN apt update && apt install -y nvtop -RUN pip install torchvision torchaudio transformers datasets fsspec==2023.9.2 python-etcd numpy==1.* -RUN pip install torch==2.2.0+cu121 --index-url https://download.pytorch.org/whl/cu121 +RUN pip install torchvision torchaudio transformers==4.46.1 datasets fsspec==2023.9.2 python-etcd numpy==1.* +RUN pip install torch==2.5.1+cu121 --index-url https://download.pytorch.org/whl/cu121 RUN mkdir /checkpoints