fix bugs

.github/workflows/docker-publish.yaml

@@ -27,13 +27,13 @@ jobs:
         include:
           - cuda_version: 12.4
             lite: true
-            tag_prefix: lite-cu124
+            tag_prefix: cu124-lite
           - cuda_version: 12.4
             lite: false
             tag_prefix: cu124
           - cuda_version: 12.8
             lite: true
-            tag_prefix: lite-cu128
+            tag_prefix: cu128-lite
           - cuda_version: 12.8
             lite: false
             tag_prefix: cu128

Docker/anaconda_install.sh

@@ -45,6 +45,8 @@ source "$HOME/anaconda3/etc/profile.d/conda.sh"
 
 "$HOME/anaconda3/bin/conda" update -q --all -y 1>/dev/null
 
+"$HOME/anaconda3/bin/conda" install python=3.11 -q -y
+
 rm $LOG_PATH
 
 rm -rf "$HOME/anaconda3/pkgs"

Docker/setup.sh

@@ -48,9 +48,13 @@ fi
 
 source "$HOME/anaconda3/etc/profile.d/conda.sh"
 
+echo "CUDA_VERSION: $CUDA_VERSION"
+
 if [ "$CUDA_VERSION" = 128 ]; then
+    echo 1111111
     pip install torch torchaudio --no-cache-dir --index-url https://download.pytorch.org/whl/cu128
 elif [ "$CUDA_VERSION" = 124 ]; then
+    echo 2222222
     pip install torch==2.5.1 torchaudio==2.5.1 --no-cache-dir --index-url https://download.pytorch.org/whl/cu124
 fi