diff --git a/requirements.txt b/requirements.txt
index e5dd21c..158e710 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,5 +1,5 @@
-torch==1.9.0
-torchvision==0.10.0
+torch
+torchvision
 pillow==8.3.1
 numpy==1.19.5
 timm==0.4.12