diff --git a/.github/workflows/docker-publish.yaml b/.github/workflows/docker-publish.yaml
index 39785290..e2be5423 100644
--- a/.github/workflows/docker-publish.yaml
+++ b/.github/workflows/docker-publish.yaml
@@ -24,19 +24,19 @@ jobs:
     strategy:
       matrix:
         include:
-          - cuda_version: 124
+          - cuda_version: 12.4
             lite: true
             torch_base: lite
             tag_prefix: cu124-lite
-          - cuda_version: 124
+          - cuda_version: 12.4
             lite: false
             torch_base: full
             tag_prefix: cu124
-          - cuda_version: 128
+          - cuda_version: 12.8
             lite: true
             torch_base: lite
             tag_prefix: cu128-lite
-          - cuda_version: 128
+          - cuda_version: 12.8
             lite: false
             torch_base: full
             tag_prefix: cu128
@@ -111,19 +111,19 @@ jobs:
     strategy:
       matrix:
         include:
-          - cuda_version: 124
+          - cuda_version: 12.4
             lite: true
             torch_base: lite
             tag_prefix: cu124-lite
-          - cuda_version: 124
+          - cuda_version: 12.4
             lite: false
             torch_base: full
             tag_prefix: cu124
-          - cuda_version: 128
+          - cuda_version: 12.8
             lite: true
             torch_base: lite
             tag_prefix: cu128-lite
-          - cuda_version: 128
+          - cuda_version: 12.8
             lite: false
             torch_base: full
             tag_prefix: cu128
diff --git a/Dockerfile b/Dockerfile
index 429d7f4c..34a6d49f 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-ARG CUDA_VERSION=124
+ARG CUDA_VERSION=12.4
 ARG TORCH_BASE=full
 
 FROM xxxxrt666/torch-base:cu${CUDA_VERSION}-${TORCH_BASE}
@@ -7,7 +7,7 @@ LABEL maintainer="XXXXRT"
 LABEL version="V4"
 LABEL description="Docker image for GPT-SoVITS"
 
-ARG CUDA_VERSION=124
+ARG CUDA_VERSION=12.4
 
 ENV CUDA_VERSION=${CUDA_VERSION}
 
diff --git a/docker_build.sh b/docker_build.sh
index ef452ce3..4db652a1 100644
--- a/docker_build.sh
+++ b/docker_build.sh
@@ -40,10 +40,10 @@ while [[ $# -gt 0 ]]; do
     --cuda)
         case "$2" in
         12.4)
-            CUDA_VERSION=124
+            CUDA_VERSION=12.4
             ;;
         12.8)
-            CUDA_VERSION=128
+            CUDA_VERSION=12.8
             ;;
         *)
             echo "Error: Invalid CUDA_VERSION: $2"