Update docker deps.

This commit is contained in:
Ruben van de Ven 2023-03-31 15:58:46 +02:00
parent abf8085ab7
commit 9aa9e9c709

View file

@ -1,5 +1,6 @@
FROM pytorch/pytorch:1.3-cuda10.1-cudnn7-devel
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