Move some settings from .bashrc to .setup_env.sh

Author: jamesnulliu

Dockerfile.cpu

@@ -61,7 +61,8 @@ RUN wget -qO- https://apt.llvm.org/llvm-snapshot.gpg.key | \
 RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
 
 # Config files
-COPY data/.vimrc data/.inputrc data/.bashrc data/.tmux.conf /root/
+COPY data/.vimrc data/.inputrc data/.bashrc data/.setup_env.sh data/.tmux.conf \
+    /root/
 
 # Install Miniconda3 and conda env
 # [TODO] Conda now uses python=3.13 in default. However, some packages (i.e.,

Dockerfile.cuda

@@ -61,7 +61,8 @@ RUN wget -qO- https://apt.llvm.org/llvm-snapshot.gpg.key | \
 RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y
 
 # Config files
-COPY data/.vimrc data/.inputrc data/.bashrc data/.tmux.conf /root/
+COPY data/.vimrc data/.inputrc data/.bashrc data/.setup_env.sh data/.tmux.conf \
+    /root/
 
 # Install Miniconda3 and conda env
 # [TODO] Conda now uses python=3.13 in default. However, some packages (i.e.,

data/.bashrc

@@ -102,56 +102,5 @@ fi
 #    . /etc/bash_completion
 #fi
 
-# >>> conda initialize >>>
-# !! Contents within this block are managed by 'conda init' !!
-__conda_setup="$('/root/miniconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
-if [ $? -eq 0 ]; then
-    eval "$__conda_setup"
-else
-    if [ -f "/root/miniconda3/etc/profile.d/conda.sh" ]; then
-        . "/root/miniconda3/etc/profile.d/conda.sh"
-    else
-        export PATH="/root/miniconda3/bin:$PATH"
-    fi
-fi
-unset __conda_setup
-# <<< conda initialize <<<
-
-# @brief Add `$1` into environment variable `$2` if it is not already there.
-# @example > env_load PATH /usr/local/bin
-env_load() {
-    local env_var=$1
-    local path=$2
-    if [[ ":${!env_var}:" != *":$path:"* ]]; then
-        export $env_var="${!env_var}:$path"
-    fi
-}
-
-# @brief Remove `$1` from environment variable `$2` if it is there.
-# @example > env_unload PATH /usr/local/bin
-env_unload() {
-    local env_var=$1
-    local path=$2
-    local paths_array=(${!env_var//:/ })
-    local new_paths=()
-    for item in "${paths_array[@]}"; do
-        if [[ "$item" != "$path" ]]; then
-            new_paths+=("$item")
-        fi
-    done
-    export $env_var=$(IFS=:; echo "${new_paths[*]}")
-}
-
-export CUDA_HOME="/usr/local/cuda"
-alias LOAD_CUDA="env_load PATH $CUDA_HOME/bin && env_load LD_LIBRARY_PATH $CUDA_HOME/lib64"
-alias UNLOAD_CUDA="env_unload PATH $CUDA_HOME/bin && env_unload LD_LIBRARY_PATH $CUDA_HOME/lib64"
-# Load CUDA by default
-LOAD_CUDA
-
-export VCPKG_HOME="/usr/local/vcpkg"
-export VCPKG_ROOT=$VCPKG_HOME
-alias LOAD_VCPKG="env_load PATH $VCPKG_HOME"
-alias UNLOAD_VCPKG="env_unload PATH $VCPKG_HOME"
-alias VCPKG_UPDATE="pushd $VCPKG_HOME && git pull && popd"
-# Load vcpkg by default
-LOAD_VCPKG
+# Load all environment settings
+source ~/.setup_env.sh

data/.setup_env.sh

@@ -0,0 +1,53 @@
+# @brief Add `$1` into environment variable `$2` if it is not already there.
+# @example > env_load PATH /usr/local/bin
+env_load() {
+    local env_var=$1
+    local path=$2
+    if [[ ":${!env_var}:" != *":$path:"* ]]; then
+        export $env_var="${!env_var}:$path"
+    fi
+}
+
+# @brief Remove `$1` from environment variable `$2` if it is there.
+# @example > env_unload PATH /usr/local/bin
+env_unload() {
+    local env_var=$1
+    local path=$2
+    local paths_array=(${!env_var//:/ })
+    local new_paths=()
+    for item in "${paths_array[@]}"; do
+        if [[ "$item" != "$path" ]]; then
+            new_paths+=("$item")
+        fi
+    done
+    export $env_var=$(IFS=:; echo "${new_paths[*]}")
+}
+
+# >>> conda initialize >>>
+# !! Contents within this block are managed by 'conda init' !!
+__conda_setup="$('/root/miniconda3/bin/conda' 'shell.bash' 'hook' 2> /dev/null)"
+if [ $? -eq 0 ]; then
+    eval "$__conda_setup"
+else
+    if [ -f "/root/miniconda3/etc/profile.d/conda.sh" ]; then
+        . "/root/miniconda3/etc/profile.d/conda.sh"
+    else
+        env_load PATH "/root/miniconda3/bin"
+    fi
+fi
+unset __conda_setup
+# <<< conda initialize <<<
+
+export CUDA_HOME="/usr/local/cuda"
+alias LOAD_CUDA="env_load PATH $CUDA_HOME/bin && env_load LD_LIBRARY_PATH $CUDA_HOME/lib64"
+alias UNLOAD_CUDA="env_unload PATH $CUDA_HOME/bin && env_unload LD_LIBRARY_PATH $CUDA_HOME/lib64"
+# Load CUDA by default
+LOAD_CUDA
+
+export VCPKG_HOME="/usr/local/vcpkg"
+export VCPKG_ROOT=$VCPKG_HOME
+alias LOAD_VCPKG="env_load PATH $VCPKG_HOME"
+alias UNLOAD_VCPKG="env_unload PATH $VCPKG_HOME"
+alias VCPKG_UPDATE="pushd $VCPKG_HOME && git pull && popd"
+# Load vcpkg by default
+LOAD_VCPKG

scripts/.image-configs.sh

@@ -1,4 +1,4 @@
-IMAGE_VERSION=2.2.3
+IMAGE_VERSION=2.2.4
 TORCH_VERSION=2.7.0
 CUDA_VERSION=12.8.0
 UBUNTU_VERSION=24.04