diff --git a/setup.py b/setup.py index 06b938bb31..8030c5408e 100644 --- a/setup.py +++ b/setup.py @@ -86,8 +86,12 @@ def _main(): print("-- Git branch:", branch) print("-- Git SHA:", sha) print("-- Git tag:", tag) - pytorch_package_dep = _get_pytorch_version() - print("-- PyTorch dependency:", pytorch_package_dep) + # This used to be passed to install_requires + # which would cause pinning against a specific torch version in releases. + # I don't think we want to pin at all? + # TODO: revisit if needed. Maybe it's needed for nightlies. Unsure. + # pytorch_package_dep = _get_pytorch_version() + # print("-- PyTorch dependency:", pytorch_package_dep) version = _get_version(sha) print("-- Building version", version) @@ -135,7 +139,7 @@ def _main(): "build_ext": setup_helpers.get_build_ext(), "clean": clean, }, - install_requires=[pytorch_package_dep], + install_requires=[], zip_safe=False, )