2.1.3

Author: jamesnulliu

Dockerfile

@@ -70,8 +70,9 @@ RUN wget -O /tmp/miniconda3.sh \
     \. /root/miniconda3/bin/activate && \
     conda upgrade libstdcxx-ng -c conda-forge -y && \
     pip3 install nvitop && \
+    TORCH_CU_VER=$(echo $CUDA_VERSION | cut -d'.' -f1,2 | tr -d '.') && \ 
     pip3 install torch==${TORCH_VERSION} torchvision torchaudio \
-        --index-url https://download.pytorch.org/whl/cu126 \
+        --index-url "https://download.pytorch.org/whl/cu${TORCH_CU_VER}" \
         --no-cache-dir
 
 # Some final steps

scripts/.image-configs.sh

@@ -1,6 +1,6 @@
-IMAGE_VERSION=2.1.2
-TORCH_VERSION=2.7.0
-CUDA_VERSION=12.6.0
+IMAGE_VERSION=2.1.3
+TORCH_VERSION=2.7.1
+CUDA_VERSION=12.8.0
 UBUNTU_VERSION=24.04
 LLVM_VERSION=21
 IMAGE_TAG=v${IMAGE_VERSION}-torch${TORCH_VERSION}-cuda${CUDA_VERSION}-ubuntu${UBUNTU_VERSION}