make pytorch dep a lower bound

pytorch · Apr 10, 2024 · 058f9ab · 058f9ab
1 parent ecb9ebc
commit 058f9ab
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -66,7 +66,7 @@ def _init_submodule():
 
 pytorch_package_dep = "torch"
 if pytorch_package_version is not None:
-    pytorch_package_dep += "==" + pytorch_package_version
+    pytorch_package_dep += ">=" + pytorch_package_version
 
 
 class clean(distutils.command.clean.clean):