2023-03-31 13:58:46 +00:00
|
|
|
FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-devel
|
2020-01-21 02:50:51 +00:00
|
|
|
|
|
|
|
|
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
|
2023-03-31 13:58:46 +00:00
|
|
|
RUN pip install lap
|
2023-04-05 15:16:28 +00:00
|
|
|
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 panel jupyter_bokeh
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# for bokeh
|
|
|
|
|
EXPOSE 5006
|
|
|
|
|
|
|
|
|
|
CMD python -m ipykernel_launcher -f $DOCKERNEL_CONNECTION_FILE
|
