diff --git a/requirements.txt b/requirements.txt
index d6be752..0c15207 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,6 +1,6 @@
 pandas>=1.3.0
 numpy
 joblib==1.3.1
-torch==2.0.1
+torch>=2.0.1
 tqdm==4.66.3
 scikit-learn==1.3.0