diff --git a/requirement.txt b/requirement.txt
index 21994f4..6d12ee0 100644
--- a/requirement.txt
+++ b/requirement.txt
@@ -1,5 +1,4 @@
-python>=3.6
-pytorch>=1.8
+torch>=1.8
 numpy
 pandas
 blobfile