blackteay/GPT-SoVITS
1
0
Fork 0
mirror of https://github.com/RVC-Boss/GPT-SoVITS.git synced 2025-09-07 00:29:48 +08:00
Code Issues Packages Projects Releases Wiki Activity

.

Browse Source
This commit is contained in:
XXXXRT666 2025-04-30 23:37:22 +01:00
parent 0ac28b42ae
commit ac5ed0ad61
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile
View File

@ -1,6 +1,7 @@
ARG CUDA_VERSION=12.4
ARG CUDA_BASE=runtime
FROM nvidia/cuda:${CUDA_VERSION}.1-cudnn-devel-ubuntu22.04
FROM nvidia/cuda:${CUDA_VERSION}.1-cudnn-${CUDA_BASE}-ubuntu22.04
LABEL maintainer="XXXXRT"
LABEL version="V4-0501"

7
docker_build.sh
View File

@ -67,9 +67,16 @@ done
TARGETPLATFORM=$(uname -m | grep -q 'x86_64' && echo "linux/amd64" || echo "linux/arm64")
if [ $LITE = true ]; then
CUDA_BASE="runtime"
else
CUDA_BASE="devel"
fi
docker build \
--build-arg CUDA_VERSION=$CUDA_VERSION \
--build-arg LITE=$LITE \
--build-arg TARGETPLATFORM="$TARGETPLATFORM" \
--build-arg CUDA_BASE=$CUDA_BASE \
-t "${USER}/gpt-sovits:local" \
.