Merge pull request #936 from huyyxy/main

feat(docker): update CUDA base image to 12.4.1 for TensorRT support
This commit is contained in:
Xiang Lyu
2025-02-06 11:39:53 +08:00
committed by GitHub

View File

@@ -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