From a315c44ab709e8e3e740614a9ba2c52b0bfe18f0 Mon Sep 17 00:00:00 2001 From: KleinYuan Date: Mon, 6 Feb 2023 16:44:46 -0800 Subject: [PATCH] add docker support --- .gitignore | 1 + Dockerfile | 22 ++++++++++++++++++++++ Makefile | 8 ++++++++ requirements.txt | 2 +- 4 files changed, 32 insertions(+), 1 deletion(-) create mode 100644 Dockerfile create mode 100644 Makefile diff --git a/.gitignore b/.gitignore index 6c093a6..b966cd9 100644 --- a/.gitignore +++ b/.gitignore @@ -1,3 +1,4 @@ +.idea/ # Byte-compiled / optimized / DLL files __pycache__/ *.py[cod] diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..b537297 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,22 @@ +FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04 + +ENV DEBIAN_FRONTEND noninteractive +RUN apt-get update && apt-get install -y \ + python3-opencv ca-certificates python3-dev git wget sudo ninja-build +RUN ln -sv /usr/bin/python3 /usr/bin/python + +COPY . /home/appuser +WORKDIR /home/appuser + +# https://github.com/facebookresearch/detectron2/issues/3933 +ENV PATH="/home/appuser/.local/bin:${PATH}" +RUN wget https://bootstrap.pypa.io/pip/3.6/get-pip.py && \ + python3 get-pip.py && \ + rm get-pip.py + +# install dependencies +# See https://pytorch.org/ for other options if you use a different version of CUDA +RUN pip install -r requirements.txt +RUN pip install torch==1.10 torchvision==0.11.1 -f https://download.pytorch.org/whl/cu111/torch_stable.html +RUN pip install setuptools==59.5.0 +RUN python setup.py develop diff --git a/Makefile b/Makefile new file mode 100644 index 0000000..53c9577 --- /dev/null +++ b/Makefile @@ -0,0 +1,8 @@ +clean: + docker rm -f $$(docker ps -qa) + +build-image: + docker build -t=deeppreid:v0 . + +run: + nvidia-docker run -v ${PWD}:/home/appuser --name=deeppreid --net=host --ipc=host -it deeppreid:v0 \ No newline at end of file diff --git a/requirements.txt b/requirements.txt index 71da6fe..f3c94ff 100644 --- a/requirements.txt +++ b/requirements.txt @@ -14,7 +14,7 @@ flake8 yapf isort==4.3.21 imageio - +chardet # Export -------------------------------------- # onnx # onnx-simplified