doc: Add context for why we add an explicit pytorch source

This commit is contained in:
ditsuke 2024-07-03 01:02:56 +05:30
parent 29e6edcc9b
commit a9d644271c
No known key found for this signature in database
GPG key ID: 71B6C31C8A5A9D21

View file

@ -26,6 +26,9 @@ torch = { version = "^2.2.0", source = "pytorch" }
[tool.poetry.dev-dependencies]
pytest = "^5.2"
# Force wheel + cpu
# For discussion and context see https://github.com/python-poetry/poetry#6409
[[tool.poetry.source]]
name = "pytorch"
url = "https://download.pytorch.org/whl/cpu"