#43: Fix pytorch install

[JushBJJ]

Allows install of either pytorch w/ cuda or plain pytorch and also fixes this error:

$ poetry lock && poetry install
Installing dependencies from lock file

Repository "pytorch" does not exist.

[citizenhicks]

this is better approach than #44 given the explicit choice between the two systems (mac or other)

[artus-LYTiQ]

With the suggested code

torch = [ {version = "^2.4.1+cu124", source = "pytorch", markers = "platform_machine != 'Darwin'"}, {version = "^2.4.1", source = "pytorch"} ] ... [[tool.poetry.source]] name = "pytorch" url = "https://download.pytorch.org/whl/cu124" priority = "explicit"

I am getting

- Updating torch (2.4.1 -> 2.4.1+cu124): Failed RuntimeError Unable to find installation candidates for torch (2.4.1+cu124) at /opt/homebrew/Cellar/poetry/1.8.3_2/libexec/lib/python3.12/site-packages/poetry/installation/chooser.py:74 in choose_for 70│ 71│ links.append(link) 72│ 73│ if not links: → 74│ raise RuntimeError(f"Unable to find installation candidates for {package}") 75│ 76│ # Get the best link 77│ chosen = max(links, key=lambda link: self._sort_key(package, link)) 78│ Cannot install torch.

I tried quite a bit but in the end I cannot install PyTorch on Mac from https://download.pytorch.org/whl/cu124 which I think is in line with the PyTorch.org installation guide.

If you dislike the verbose sys_platform or machine_... syntax, then this here did work on my Mac, too. Did not test again on Ubuntu, though.

torch = [ {version = "^2.4.1", platform = "darwin", source = "pypi"}, {version = "^2.4.1+cu124", source = "pytorch", platform = "linux"} ]

[JushBJJ]

Thanks for telling me this, i was hoping that in MacOS that poetry would switch to the non-cuda version of torch. I'll test it out now

[JushBJJ]

We can probably do that change, but will ignore CPU users in other platforms. Might be reasonable for now until we can figure out how to make install CPU torch by default then just do a flag (e.g --extras) with cuda

[JushBJJ]

shrek is gonna switch from poetry to uv soon
https://x.com/_xjdr/status/1845525974143943093?t=Wy_wGfRFgy-Px3tEVaim_g&s=19

[clxyder]

I can't believe that just worked