diff --git a/requirements.txt b/requirements.txt
index 289ca08..f6f8b44 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
 ftfy
 regex
 tqdm
-torch>=1.7.1,<1.7.2
-torchvision==0.8.2
+torch~=1.7.1
+torchvision~=0.8.2