diff --git a/docker/Dockerfile b/docker/Dockerfile
index 8f88df67..d77be958 100644
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
@@ -1,4 +1,4 @@
-FROM pytorch/pytorch:1.8.1-cuda11.1-cudnn8-devel
+FROM pytorch/pytorch:1.10.0-cuda11.3-cudnn8-devel
 
 RUN apt-get update \
     && apt-get -y install libgl1-mesa-glx \