diff --git a/.github/workflows/docker-publish.yaml b/.github/workflows/docker-publish.yaml
index 20dcea89..767a4743 100644
--- a/.github/workflows/docker-publish.yaml
+++ b/.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
diff --git a/Docker/anaconda_install.sh b/Docker/anaconda_install.sh
index 3b578ad0..da916714 100644
--- a/Docker/anaconda_install.sh
+++ b/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"
diff --git a/Docker/setup.sh b/Docker/setup.sh
index 9653be42..381d7e40 100644
--- a/Docker/setup.sh
+++ b/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