remove nvidia-cudnn-cu12

supervisely-ecosystem · Oct 25, 2024 · 9479a63 · 9479a63
1 parent 87ca83c
commit 9479a63
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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