Update v2.1.3

Author: jamesnulliu

Dockerfile

@@ -10,6 +10,7 @@ ENV LANG=en_US.UTF-8
 
 ARG IMAGE_VERSION
 ARG TORCH_VERSION
+ARG TORCH_CUDA_VERSION
 
 LABEL maintainer="JamesNULLiu jamesnulliu@gmail.com"
 LABEL version=${IMAGE_VERSION}
@@ -21,25 +22,34 @@ RUN apt-get update && apt-get upgrade -y && \
     apt-get install -y \
     apt-utils lsb-release software-properties-common gnupg git \
     vim-gtk3 wget p7zip-full ninja-build curl jq pkg-config ssh \
-    build-essential gdb htop tmux ttf-mscorefonts-installer  kmod \
+    build-essential gdb htop tmux ttf-mscorefonts-installer kmod \
     libssl-dev && \
     fc-cache -f -v
 
-# Vcpkg, Cmake, LLVM
+# Vcpkg
 RUN cd /usr/local && git clone https://github.com/microsoft/vcpkg.git && \ 
-    cd vcpkg && ./bootstrap-vcpkg.sh  && \
-    wget -O /tmp/kitware-archive.sh \
-    https://apt.kitware.com/kitware-archive.sh && \
+    cd vcpkg && ./bootstrap-vcpkg.sh
+
+# Cmake
+RUN wget -O /tmp/kitware-archive.sh \
+        https://apt.kitware.com/kitware-archive.sh && \
     bash /tmp/kitware-archive.sh && \
-    apt-get update && apt-get install -y cmake && \
-    wget -O /tmp/llvm.sh https://apt.llvm.org/llvm.sh && \
-    chmod +x /tmp/llvm.sh && /tmp/llvm.sh ${LLVM_VERSION} && \
-    apt-get install -y libomp-${LLVM_VERSION}-dev && \
-    ln -s /usr/bin/clang-${LLVM_VERSION} /usr/bin/clang && \
-    ln -s /usr/bin/clang++-${LLVM_VERSION} /usr/bin/clang++ && \
-    ln -s /usr/bin/clangd-${LLVM_VERSION} /usr/bin/clangd && \
-    ln -s /usr/bin/clang-tidy-${LLVM_VERSION} /usr/bin/clang-tidy && \
-    ln -s /usr/bin/clang-format-${LLVM_VERSION} /usr/bin/clang-format
+    apt-get update && apt-get install -y cmake 
+
+# LLVM
+RUN wget -qO- https://apt.llvm.org/llvm-snapshot.gpg.key | \
+        apt-key add - && \
+    echo "deb http://apt.llvm.org/noble/ llvm-toolchain-noble main" | \
+        tee /etc/apt/sources.list.d/llvm.list && \
+    echo "deb-src http://apt.llvm.org/noble/ llvm-toolchain-noble main" | \
+        tee -a /etc/apt/sources.list.d/llvm.list && \
+    apt-get update && apt-get install -y clang-format clang-tidy clang-tools \
+        clang clangd libc++-dev libc++1 libc++abi-dev libc++abi1 libclang-dev \
+        libclang1 liblldb-dev libllvm-ocaml-dev libomp-dev libomp5 lld lldb \
+        llvm-dev llvm-runtime llvm python3-clang 
+
+# Rust
+RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -y
 
 # Config files
 COPY data/.vimrc data/.inputrc data/.bashrc data/.tmux.conf /root/
@@ -51,8 +61,9 @@ 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 torch==${TORCH_VERSION} torchvision torchaudio \
-        --index-url https://download.pytorch.org/whl/cu126 \
+        --index-url "https://download.pytorch.org/whl/${TORCH_CUDA_VERSION}" \
         --no-cache-dir
 
 # Some final steps

scripts/.image-configs.sh

@@ -1,6 +1,7 @@
-IMAGE_VERSION=2.1.1
-TORCH_VERSION=2.7.0
-CUDA_VERSION=12.6.0
+IMAGE_VERSION=2.1.2
+TORCH_VERSION=2.7.1
+CUDA_VERSION=12.8.0
+TORCH_CUDA_VERSION=cu128
 UBUNTU_VERSION=24.04
 IMAGE_TAG=v${IMAGE_VERSION}-torch${TORCH_VERSION}-cuda${CUDA_VERSION}-ubuntu${UBUNTU_VERSION}
 IMAGE_NAME=jamesnulliu/deeplearning:${IMAGE_TAG}