From fd5b1a31c5e9415d57992cbfb8010ebac3004491 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Sat, 1 Apr 2023 19:08:33 +0000 Subject: [PATCH] update docker --- docker/Dockerfile | 2 +- docker/requirements.txt | 4 +++- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/docker/Dockerfile b/docker/Dockerfile index cd5d9f3..3060e57 100644 --- a/docker/Dockerfile +++ b/docker/Dockerfile @@ -1,5 +1,5 @@ # Use an existing image as the base -FROM nvcr.io/nvidia/pytorch:23.03-py3 +FROM nvcr.io/nvidia/pytorch:21.10-py3 # Add the current user to the image ARG USER_NAME=lhunor diff --git a/docker/requirements.txt b/docker/requirements.txt index a7754b8..2f3990c 100644 --- a/docker/requirements.txt +++ b/docker/requirements.txt @@ -5,4 +5,6 @@ wandb scipy plotly pandas -cmapy \ No newline at end of file +cmapy +pytorch3d +tables \ No newline at end of file