make pytorch dep a lower bound (#2256)

pytorch · Apr 10, 2024 · f3b7a01 · f3b7a01
1 parent 4bf6b30
commit f3b7a01
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):