pyg-lib icon indicating copy to clipboard operation
pyg-lib copied to clipboard

Specify compatible PyTorch version in Requires-Dist

Open mikepii opened this issue 1 year ago • 2 comments

🚀 The feature, motivation and pitch

Since each distribution wheel is built for a particular version of PyTorch, it would be helpful to specify that PyTorch version in Requires-Dist in the package metadata. E.g., in setup.py: setup(install_requires=['torch==1.13.1+cu116']).

This helps fail-fast if an incompatible version is being installed. For organizations storing these wheels in a private package repository, this also allows pip to automatically select the right version if a version range is specified.

Alternatives

Can otherwise specify a particular Torch version's subdirectory of the downloads site using pip --find-links ..., but that doesn't fail-fast if the version is incompatible

Additional context

No response

mikepii avatar Feb 15 '23 21:02 mikepii