[update] Enhance scripts && Remove System Python

Author: jamesnulliu

Dockerfile.cpu

@@ -77,7 +77,10 @@ RUN curl -LsSf https://astral.sh/uv/install.sh | \
         env UV_INSTALL_DIR="${UV_HOME}" UV_NO_MODIFY_PATH=1 sh
 
 # Some final steps
-RUN apt-get update && apt-get upgrade -y && apt-get autoremove -y && \
+RUN apt-get update && apt-get upgrade -y && \
+    # Remove system python3.12, force to use uv managed python
+    apt-get remove -y python3 python3.12 && \
+    apt-get autoremove -y && \
     apt-get clean && rm -rf /var/lib/apt/lists/* && \
     git config --system --unset-all user.name  || true && \
     git config --system --unset-all user.email || true && \

Dockerfile.cuda

@@ -77,7 +77,10 @@ RUN curl -LsSf https://astral.sh/uv/install.sh | \
         env UV_INSTALL_DIR="${UV_HOME}" UV_NO_MODIFY_PATH=1 sh
 
 # Some final steps
-RUN apt-get update && apt-get upgrade -y && apt-get autoremove -y && \
+RUN apt-get update && apt-get upgrade -y && \
+    # Remove system python3.12, force to use uv managed python
+    apt-get remove -y python3 python3.12 && \
+    apt-get autoremove -y && \
     apt-get clean && rm -rf /var/lib/apt/lists/* && \
     git config --system --unset-all user.name  || true && \
     git config --system --unset-all user.email || true && \

data/create-shared-dir

@@ -21,17 +21,23 @@ set -e
 #
 # ==============================================================================
 
-# --- Argument Parsing ---
-SHARED_GROUP="$1"
-shift # Shift arguments so $@ contains only the directories
-TARGET_DIRS=("$@")
-
 # --- Input Validation ---
-if [ ${#TARGET_DIRS[@]} -eq 0 ] || [ "$1" == "-h" ] || [ "$1" == "--help" ]; then
-    echo "Usage: $0 <shared_group> <dir1> [<dir2> ...]"
+if [ $# -eq 0 ] || [ "$1" == "-h" ] || [ "$1" == "--help" ]; then
+    echo "Usage: $0 <user_group> <dir1> [<dir2> ...]"
     exit 1
 fi
 
+if [ $# -lt 2 ]; then
+    echo "Error: Missing directory arguments."
+    echo "Usage: $0 <user_group> <dir1> [<dir2> ...]"
+    exit 1
+fi
+
+# --- Argument Parsing ---
+USER_GROUP="$1"
+shift
+TARGET_DIRS=("$@")
+
 # --- Pre-flight Checks ---
 # 1. Check for root privileges.
 if [[ $EUID -ne 0 ]]; then
@@ -40,22 +46,22 @@ if [[ $EUID -ne 0 ]]; then
 fi
 
 # 2. Check if the specified group exists.
-if ! getent group "$SHARED_GROUP" > /dev/null; then
-    echo "Error: Group '$SHARED_GROUP' does not exist." >&2
+if ! getent group "$USER_GROUP" > /dev/null; then
+    echo "Error: Group '$USER_GROUP' does not exist." >&2
     exit 1
 fi
 
 # --- Main Logic ---
-echo "Configuring shared directories for group '$SHARED_GROUP'..."
+echo "Configuring shared directories for group '$USER_GROUP'..."
 
 for target_dir in "${TARGET_DIRS[@]}"; do
     echo "--> Processing: $target_dir"
     mkdir -p "$target_dir"
-    chgrp -R "$SHARED_GROUP" "$target_dir"
+    chgrp -R "$USER_GROUP" "$target_dir"
     chmod -R g=rwX,o-rwx "$target_dir"
     find "$target_dir" -type d -exec chmod g+s {} +
-    setfacl -R -m "g:$SHARED_GROUP:rwX" "$target_dir"
-    setfacl -R -d -m "g:$SHARED_GROUP:rwX" "$target_dir"
+    setfacl -R -m "g:$USER_GROUP:rwX" "$target_dir"
+    setfacl -R -d -m "g:$USER_GROUP:rwX" "$target_dir"
 done
 
 echo "Configuration complete."

data/env_setup.sh

@@ -70,23 +70,23 @@ if [ -d "${UV_HOME:-}" ]; then
     alias UNLOAD_UV="env_unload PATH $UV_HOME"
     env_load PATH "$UV_HOME"
     echo "[ENV-SETUP] UV initialized from $UV_HOME"
-    cat << 'EOF'
-|- NOTE: For managing shared UV cache and python directory:
-|-   1. Export UV_CACHE_DIR and UV_PYTHON_INSTALL_DIR to a target directory. 
-|       (No need to create them first.)
-|-   2. Run command below ONLY ONCE to create the shared directories: 
-|       $ create-shared-dir <group> ${UV_CACHE_DIR} ${UV_PYTHON_INSTALL_DIR}
-|-   3. For all users in <group>, make sure UV_CACHE_DIR and 
-|       UV_PYTHON_INSTALL_DIR are exported properly.
-|       You can do this in ${ENV_SETUP_FILE} for convenience.
-|-   4. Remove this NOTE from ${ENV_SETUP_FILE} if you want.
-EOF
     ## You can uncomment the following line to set a shared cache dir for UV.
     ## Make sure to run `sudo create-shared-dir <group> $UV_CACHE_DIR` ONLY ONCE.
     # export UV_CACHE_DIR="$UV_HOME/.cache"
     # export UV_PYTHON_INSTALL_DIR="$UV_HOME/python"
-    echo "|-          UV_CACHE_DIR: ${UV_CACHE_DIR:-<not set>}"
-    echo "|-          UV_PYTHON_INSTALL_DIR: ${UV_PYTHON_INSTALL_DIR:-<not set>}"
+    echo "            |- UV_CACHE_DIR: ${UV_CACHE_DIR:-<not set>}"
+    echo "            |- UV_PYTHON_INSTALL_DIR: ${UV_PYTHON_INSTALL_DIR:-<not set>}"
+    cat << 'EOF'
+   NOTE: For managing shared UV cache and python directory:
+     1. Export UV_CACHE_DIR and UV_PYTHON_INSTALL_DIR to a target directory. 
+        (No need to create them first.)
+     2. Run command below ONLY ONCE to create the shared directories: 
+        $ create-shared-dir <group> ${UV_CACHE_DIR} ${UV_PYTHON_INSTALL_DIR}
+     3. For all users in <group>, make sure UV_CACHE_DIR and 
+        UV_PYTHON_INSTALL_DIR are exported properly.
+        You can do this in ${ENV_SETUP_FILE} for convenience.
+     4. Remove this NOTE from ${ENV_SETUP_FILE} if you want.
+EOF
 else
     unset UV_HOME
 fi

scripts/build.sh

@@ -18,9 +18,9 @@ docker build \
     -f $DOCKER_FILE \
     --build-arg IMAGE_VERSION=$IMAGE_VERSION \
     --build-arg IMAGE_NAME=$IMAGE_NAME \
-    --build-arg CUDA_VERSION=$CUDA_VERSION \
     --build-arg UBUNTU_VERSION=$UBUNTU_VERSION \
     --build-arg LLVM_VERSION=$LLVM_VERSION \
+    --build-arg CUDA_VERSION=$CUDA_VERSION \
     -t $IMAGE_NAME \
     .
 

scripts/image-configs.sh

@@ -1,5 +1,4 @@
-IMAGE_VERSION=2.4.3
-TORCH_VERSION=2.7.1
+IMAGE_VERSION=2.4.4
 CUDA_VERSION=12.8.0
 UBUNTU_VERSION=24.04
 LLVM_VERSION=20