From 9aa9e9c709e28c70c55eae3346122970b4f701e1 Mon Sep 17 00:00:00 2001 From: Ruben van de Ven Date: Fri, 31 Mar 2023 15:58:46 +0200 Subject: [PATCH] Update docker deps. --- docker/Dockerfile | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index c5fe6fa..f5f6c75 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -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