Update CUDA_HOME in env

Author: jamesnulliu

data/.bashrc

@@ -142,6 +142,12 @@ env_unload() {
     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"

scripts/.image-configs.sh

@@ -1,4 +1,4 @@
-IMAGE_VERSION=2.0.5
+IMAGE_VERSION=2.0.6
 TORCH_VERSION=2.7.0
 CUDA_VERSION=12.6.0
 UBUNTU_VERSION=24.04