From 79190d44e4f8ae6de94892ca2b23436f1f8b9ca1 Mon Sep 17 00:00:00 2001 From: Anthony Miyaguchi Date: Tue, 26 Sep 2023 21:19:53 -0700 Subject: [PATCH 1/2] fix: bump docker nvidia/cuda to 11.7.1 --- docker/Dockerfile | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index 6b0ee92..ab7cd02 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,4 +1,4 @@ -FROM nvidia/cuda:11.7.0-devel-ubuntu20.04 +FROM nvidia/cuda:11.7.1-devel-ubuntu20.04 # NOTE: We need cuda devel to build DCNv2, but we don't necessarily need it to # run inference. This should probably become a multi-part build if we care about From a8833790b08cc1eb96ec479d3d34c005fc917c4a Mon Sep 17 00:00:00 2001 From: Anthony Miyaguchi Date: Tue, 26 Sep 2023 21:57:25 -0700 Subject: [PATCH 2/2] fix: pin matplotlib in dockerfile --- docker/Dockerfile | 3 +++ 1 file changed, 3 insertions(+) diff --git a/docker/Dockerfile b/docker/Dockerfile index ab7cd02..18bb8e0 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -37,6 +37,9 @@ RUN apt-get update && \ RUN pip install --upgrade pip wheel cython +# As of matplotlib 3.8.0, Python3.8 is no longer supported, so install our pinned version +RUN pip install matplotlib==3.7.1 + RUN apt update && \ apt install -y --no-install-recommends \ # install ros packages