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