4 ms·
You can install pytorch with it, I have been doing so for a while. [tool.poetry.dependencies] torch = { version = "1.13.1+cu116", source = "pytorch-cuda" }
by hereonout2 3y ago
You can install pytorch with it, I have been doing so for a while.
[tool.poetry.dependencies]
torch = { version = "1.13.1+cu116", source = "pytorch-cuda" }
[[tool.poetry.source]]
name = "pytorch-cuda"
url = "https://download.pytorch.org/whl/cu116"
priority = "explicit"
But, it does seem to go off and download many GBs of pytorch packages for all possible python, architecture and OS versions which takes ages. Feels very broken at least.