21 lines
602 B
Docker
21 lines
602 B
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 panel jupyter_bokeh
|
|
|
|
|
|
# for bokeh
|
|
EXPOSE 5006
|
|
|
|
CMD python -m ipykernel_launcher -f $DOCKERNEL_CONNECTION_FILE
|