diff --git a/requirements.txt b/requirements.txt
index 775de178..ec974194 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -3,4 +3,5 @@ tensorflow-gpu == 2.4.1
 Flask == 1.1.2
 Flask-SocketIO == 5.0.1
 requests == 2.25.1
-easygui == 0.98.2
\ No newline at end of file
+easygui == 0.98.2
+torch == 1.8.1+cu111
\ No newline at end of file