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}