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:
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

4
Dockerfile
View File

@ -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}

4
docker_build.sh
View File

@ -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"