diff --git a/setup.py b/setup.py index a3fb2707c3..01810aa5c2 100644 --- 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):