blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-08-20 10:18:32 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Wrong Cuda Version

Browse Source
This commit is contained in:
XXXXRT666 2025-05-09 07:26:54 +01:00
parent dfea2563c0
commit 96e94d8666
3 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
.github/workflows/docker-publish.yaml vendored
View File

@ -24,19 +24,19 @@ jobs:
strategy: strategy:
matrix: matrix:
include: include:
- cuda_version: 124 - cuda_version: 12.4
lite: true lite: true
torch_base: lite torch_base: lite
tag_prefix: cu124-lite tag_prefix: cu124-lite
- cuda_version: 124 - cuda_version: 12.4
lite: false lite: false
torch_base: full torch_base: full
tag_prefix: cu124 tag_prefix: cu124
- cuda_version: 128 - cuda_version: 12.8
lite: true lite: true
torch_base: lite torch_base: lite
tag_prefix: cu128-lite tag_prefix: cu128-lite
- cuda_version: 128 - cuda_version: 12.8
lite: false lite: false
torch_base: full torch_base: full
tag_prefix: cu128 tag_prefix: cu128
@ -111,19 +111,19 @@ jobs:
strategy: strategy:
matrix: matrix:
include: include:
- cuda_version: 124 - cuda_version: 12.4
lite: true lite: true
torch_base: lite torch_base: lite
tag_prefix: cu124-lite tag_prefix: cu124-lite
- cuda_version: 124 - cuda_version: 12.4
lite: false lite: false
torch_base: full torch_base: full
tag_prefix: cu124 tag_prefix: cu124
- cuda_version: 128 - cuda_version: 12.8
lite: true lite: true
torch_base: lite torch_base: lite
tag_prefix: cu128-lite tag_prefix: cu128-lite
- cuda_version: 128 - cuda_version: 12.8
lite: false lite: false
torch_base: full torch_base: full
tag_prefix: cu128 tag_prefix: cu128

4
Dockerfile
View File

@ -1,4 +1,4 @@
ARG CUDA_VERSION=124 ARG CUDA_VERSION=12.4
ARG TORCH_BASE=full ARG TORCH_BASE=full
FROM xxxxrt666/torch-base:cu${CUDA_VERSION}-${TORCH_BASE} FROM xxxxrt666/torch-base:cu${CUDA_VERSION}-${TORCH_BASE}
@ -7,7 +7,7 @@ LABEL maintainer="XXXXRT"
LABEL version="V4" LABEL version="V4"
LABEL description="Docker image for GPT-SoVITS" LABEL description="Docker image for GPT-SoVITS"
ARG CUDA_VERSION=124 ARG CUDA_VERSION=12.4
ENV CUDA_VERSION=${CUDA_VERSION} ENV CUDA_VERSION=${CUDA_VERSION}

4
docker_build.sh
View File

@ -40,10 +40,10 @@ while [[ $# -gt 0 ]]; do
--cuda) --cuda)
case "$2" in case "$2" in
12.4) 12.4)
CUDA_VERSION=124 CUDA_VERSION=12.4
;; ;;
12.8) 12.8)
CUDA_VERSION=128 CUDA_VERSION=12.8
;; ;;
*) *)
echo "Error: Invalid CUDA_VERSION: $2" echo "Error: Invalid CUDA_VERSION: $2"