31 lines
1.1 KiB
Docker
31 lines
1.1 KiB
Docker
FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-devel
|
|
|
|
RUN apt update && apt install -y ffmpeg libsm6 libxrender-dev
|
|
RUN pip install Cython
|
|
RUN pip install opencv-python cython_bbox motmetrics numba matplotlib sklearn
|
|
RUN pip install lap
|
|
RUN pip install umap-learn
|
|
ENV NUMBA_CACHE_DIR=/tmp/numba_cache
|
|
RUN pip install bokeh
|
|
RUN pip install ipykernel
|
|
RUN pip install ipython
|
|
# Vscode bug: https://github.com/microsoft/vscode-jupyter/issues/8552
|
|
# RUN pip install ipywidgets==7.7.2
|
|
RUN pip install ipywidgets
|
|
#RUN pip install panel jupyter_bokeh
|
|
|
|
# from https://github.com/KaiyangZhou/deep-person-reid/blob/master/Dockerfile
|
|
ENV DEBIAN_FRONTEND=noninteractive
|
|
ENV TZ=Europe/Amsterdam
|
|
RUN apt update && apt install -y python3-opencv ca-certificates python3-dev git wget sudo ninja-build
|
|
RUN git clone https://github.com/KaiyangZhou/deep-person-reid/ /deep-person-reid
|
|
WORKDIR /deep-person-reid
|
|
RUN pip install -r requirements.txt
|
|
RUN python setup.py develop
|
|
|
|
# for bokeh
|
|
EXPOSE 5006
|
|
|
|
ENV TORCH_HOME=/Towards-Realtime-MOT/.torch
|
|
WORKDIR /Towards-Realtime-MOT
|
|
CMD python -m ipykernel_launcher -f $DOCKERNEL_CONNECTION_FILE
|