diff --git a/Dockerfile b/Dockerfile
index 1e40a4e6..fac73803 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:2.0.0-cuda11.7-cudnn8-runtime
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 RUN ["apt", "update"]
 RUN ["apt", "install", "-y", "build-essential"]
 RUN ["pip", "install", "-U", "pip", "setuptools", "wheel"]