update docker

2023-04-01 19:08:33 +00:00
parent 03d7e3717e
commit fd5b1a31c5
2 changed files with 4 additions and 2 deletions
@@ -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
@@ -5,4 +5,6 @@ wandb
 scipy
 plotly
 pandas
-cmapy
+cmapy
+pytorch3d
+tables