Update Dockerfile

This commit is contained in:
XXXXRT666 2025-05-22 17:23:11 +03:00 committed by GitHub
parent 43de4e17d1
commit 8ddd3667d0
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -1,4 +1,4 @@
ARG CUDA_VERSION=12.4
ARG CUDA_VERSION=12.6
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=12.4
ARG CUDA_VERSION=12.6
ENV CUDA_VERSION=${CUDA_VERSION}