v2.1.3

Author: jamesnulliu

.github/workflows/cd-docker-build-push.yml

@@ -10,6 +10,17 @@ jobs:
     runs-on: ubuntu-latest
 
     steps:
+      - name: Free Disk Space (Ubuntu)
+        uses: jlumbroso/free-disk-space@main
+        with:
+          tool-cache: false
+          android: true
+          dotnet: true
+          haskell: true
+          large-packages: true
+          docker-images: false
+          swap-storage: true
+
       - name: Checkout repository
         uses: actions/checkout@v3
 

Dockerfile

@@ -59,6 +59,10 @@ RUN wget -qO- https://apt.llvm.org/llvm-snapshot.gpg.key | \
 # Rust
 RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
 
+# Clean caches here for less disk space usage
+RUN apt-get update && apt-get upgrade -y && apt-get autoremove -y && \
+    apt-get clean && rm -rf /var/lib/apt/lists/* 
+
 # Config files
 COPY data/.vimrc data/.inputrc data/.bashrc data/.tmux.conf /root/
 
@@ -69,9 +73,10 @@ RUN wget -O /tmp/miniconda3.sh \
     bash /tmp/miniconda3.sh -b -u -p /root/miniconda3 && \
     \. /root/miniconda3/bin/activate && \
     conda upgrade libstdcxx-ng -c conda-forge -y && \
-    pip3 install nvitop && \
+    pip3 install nvitop --no-cache-dir && \
+    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}