diff --git a/docker/requirements.txt b/docker/requirements.txt
index 2206d4d..6395550 100644
--- a/docker/requirements.txt
+++ b/docker/requirements.txt
@@ -19,4 +19,4 @@ pytest
 psutil
 seaborn
 tqdm
-torch==1.3.1
+torch==2.2.0