feat(docker): update CUDA base image to 12.4.1 for TensorRT support

huyyxy · huyyxy · commit aa65200713fe · 2025-01-26T12:33:50.000+08:00
- Upgrade base image from nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 to nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04 - Enable CUDA 12.4 runtime environment - Ensure TensorRT dependency compatibility - Validation steps: - Verify CUDA version via nvidia-smi after build - Test import tensorrt in container without errors Closes #935
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

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-FROM nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04`
	`1`	`+FROM nvidia/cuda:12.4.1-cudnn-devel-ubuntu22.04`
`2`	`2`
`3`	`3`	`ARG VENV_NAME="cosyvoice"`
`4`	`4`	`ENV VENV=$VENV_NAME`