heat icon indicating copy to clipboard operation
heat copied to clipboard

[Bug]: Support special versions of pytorch in setup.py

Open mtar opened this issue 2 years ago • 1 comments

What happened?

Special versions of pytorch adds a plus sign and some text to the version number like 1.12.1+cpu or 1.12.1+cu116. When installing Heat, these packages get replaced (1.12.1).

https://github.com/helmholtz-analytics/heat/blob/5ff03147ae0ff338a07efaefaf3f288e7090c6cc/setup.py#L36

Version

main (development branch), release

PyTorch version

1.12.1

mtar avatar Sep 27 '22 09:09 mtar

using less rather than less equal for the upper bound fixes the issue for me

mtar avatar Sep 27 '22 10:09 mtar