From 8ddd3667d0bfc0e3d1838f273248ed8263adbd14 Mon Sep 17 00:00:00 2001
From: XXXXRT666 <157766680+XXXXRT666@users.noreply.github.com>
Date: Thu, 22 May 2025 17:23:11 +0300
Subject: [PATCH] Update Dockerfile

---
 Dockerfile | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/Dockerfile b/Dockerfile
index 34a6d49f..71bf6fa1 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -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}