diff --git a/docs/requirements.txt b/docs/requirements.txt
index cd095e4cd3d6e7c8a5e3728d83434dc34b027337..19c357e1aa270507d029d65a4b78b0c45d31b65a 100644
--- a/docs/requirements.txt
+++ b/docs/requirements.txt
@@ -1,3 +1,3 @@
-torch
+pytorch
 torchvision
 matplotlib
\ No newline at end of file