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