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