From 96e94d8666ab7e50a5fcb7921b2947465b65cbaf Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Fri, 9 May 2025 07:26:54 +0100
Subject: [PATCH] Fix Wrong Cuda Version

---
 .github/workflows/docker-publish.yaml | 16 ++++++++--------
 Dockerfile                            |  4 ++--
 docker_build.sh                       |  4 ++--
 3 files changed, 12 insertions(+), 12 deletions(-)

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"