diff --git a/docker/Dockerfile b/docker/Dockerfile
index a2cdf54..96f3d55 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,6 +1,6 @@
 # Use an existing image as the base
 # FROM nvcr.io/nvidia/pytorch:23.04-py3
-FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-devel
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-devel
 # FROM pytorch/pytorch:1.12.0-cuda11.3-cudnn8-devel
 
 # Add the current user to the image
diff --git a/docker/docker-compose.yml b/docker/docker-compose.yml
index 240cd90..d34bde0 100644
--- a/docker/docker-compose.yml
+++ b/docker/docker-compose.yml
@@ -40,8 +40,8 @@ services:
             - capabilities: [gpu]
 
   lhunor_torch_3:
-    image: lhunor_reconstruction:v2.0
-    container_name: lhunor_reconstruction_v2.0
+    image: lhunor_reconstruction:v2.0.1
+    container_name: lhunor_reconstruction_v2.0.1
     ipc: host
     tty: true
     ulimits: