Merge pull request #936 from huyyxy/main

feat(docker): update CUDA base image to 12.4.1 for TensorRT support
FunAudioLLM · Feb 6, 2025 · da3f129 · da3f129
2 parents 86e26f5 + aa65200
commit da3f129
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04
+FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04
 
 ARG VENV_NAME="cosyvoice"
 ENV VENV=$VENV_NAME