1
0
Эх сурвалжийг харах

Merge pull request #936 from huyyxy/main

feat(docker): update CUDA base image to 12.4.1 for TensorRT support
Xiang Lyu 1 жил өмнө
parent
commit
da3f129977
1 өөрчлөгдсөн 1 нэмэгдсэн , 1 устгасан
  1. 1 1
      docker/Dockerfile

+ 1 - 1
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