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