From 9479a634e796027525df2360a2539610b7aceae2 Mon Sep 17 00:00:00 2001 From: cxnt Date: Fri, 25 Oct 2024 21:56:55 +0400 Subject: [PATCH] remove nvidia-cudnn-cu12 --- docker/Dockerfile | 1 - 1 file changed, 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 0e8b3fd4..c0a25168 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -4,7 +4,6 @@ RUN apt-get update RUN apt-get install -y curl RUN apt install -y nvidia-cuda-toolkit -RUN pip3 install -U nvidia-cudnn-cu12==8.9.7.29 RUN pip3 install torch==2.0.1 torchvision==0.15.2 --index-url https://download.pytorch.org/whl/cu117 RUN pip3 uninstall onnxruntime -y RUN pip3 install onnx==1.14.0 onnxruntime-gpu==1.15.1