From 42a7ad8dbdf8eb7e42575d26d22c4bfd9489426b Mon Sep 17 00:00:00 2001 From: Marcel Pfeiffer Date: Thu, 27 Apr 2023 22:05:08 +0200 Subject: [PATCH 1/2] Update base image to nvidia/cuda:12.0.1-cudnn8-runtime-ubuntu22.04 --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index 466bc94b..a1413183 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,7 @@ # enables cuda support in docker -FROM nvidia/cuda:10.2-cudnn7-runtime-ubuntu18.04 +FROM nvidia/cuda:12.0.1-cudnn8-runtime-ubuntu22.04 -# install python 3.6, pip and requirements for opencv-python +# install python 3.6, pip and requirements for opencv-python # (see https://github.com/NVIDIA/nvidia-docker/issues/864) RUN apt-get update && apt-get -y install \ python3 \ From b9f50a7efd5b1f426a4b702f3dba6871e3a9668e Mon Sep 17 00:00:00 2001 From: Marcel Pfeiffer Date: Thu, 27 Apr 2023 22:07:05 +0200 Subject: [PATCH 2/2] Dockerfile: Install imutils --- Dockerfile | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/Dockerfile b/Dockerfile index a1413183..7a1fa0cb 100644 --- a/Dockerfile +++ b/Dockerfile @@ -1,7 +1,7 @@ # enables cuda support in docker FROM nvidia/cuda:12.0.1-cudnn8-runtime-ubuntu22.04 -# install python 3.6, pip and requirements for opencv-python +# install python 3, pip and requirements for opencv-python # (see https://github.com/NVIDIA/nvidia-docker/issues/864) RUN apt-get update && apt-get -y install \ python3 \ @@ -14,7 +14,7 @@ RUN apt-get update && apt-get -y install \ # install python dependencies RUN pip3 install --upgrade pip -RUN pip3 install torch~=1.8 torchvision opencv-python-headless~=3.4 timm +RUN pip3 install torch~=1.8 torchvision opencv-python-headless~=3.4 timm imutils # copy inference code WORKDIR /opt/MiDaS