Poetry and PyTorch
Currently, Poetry doesn't have a -f
option (there's an open issue and an open PR), so you can't use the pip
instructions. You can install the .whl
files directly:
poetry add https://download.pytorch.org/whl/torch_stable.html/cpu/torch-1.3.1%2Bcpu-cp36-cp36m-win_amd64.whl
or add the dependency directly to your .toml
file:
[tool.poetry.dependencies]
torch = { url = "https://download.pytorch.org/whl/torch_stable.html/cpu/torch-1.3.1%2Bcpu-cp36-cp36m-win_amd64.whl" }
An updated solution from this issue in the Poetry github:
poetry add torch --platform linux --python "^3.7"
After spending a couple of hours on this issue, I found a "solution" by combining Poetry and pip just for PyTorch. You don't need to specify the wheel URLs directly and thus remain cross-platform.
I'm using Poe The Poet, a nice task runner for Poetry that allows to run any arbitrary command.
[tool.poetry.dev-dependencies]
poethepoet = "^0.10.0"
[tool.poe.tasks]
force-cuda11 = "python -m pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html"
You can run:
poetry install
and then:
poe force-cuda11 # relies on pip and use PyTorch wheels repo