Не удается установить torchvision через поэзию в Windows

Мне удалось установить pytorch благодаря ответам здесь Poetry и PyTorch.

Но установить torchvision через стихи мне так и не удалось.

> poetry add torchvision==0.8.2

Updating dependencies
Resolving dependencies...

Writing lock file

Package operations: 1 install, 0 updates, 0 removals

  • Installing torchvision (0.8.2)

  RuntimeError

  Unable to find installation candidates for torchvision (0.8.2)

  at ~\.poetry\lib\poetry\installation\chooser.py:72 in choose_for
       68│ 
       69│             links.append(link)
       70│
       71│         if not links:
    →  72│             raise RuntimeError(
       73│                 "Unable to find installation candidates for {}".format(package)
       74│             )
       75│
       76│         # Get the best link


Failed to add packages, reverting the pyproject.toml file to its original content.

Я погуглил и нашел несколько ответов, в которых говорится «просто pip install torchvision». Но я подозреваю, что это работает, потому что согласно PyPi(https://pypi.org/project/torchvision/#files), для Windows нет файла колеса. И я попробовал это, и это не удалось, как я и ожидал.

Есть ли способ установить последнюю версию torchvisin, совместимую с последней версией torch (1.7.1) в Windows? + через поэзию?


python pytorch python-poetry
person 12412316    schedule 15.12.2020    source источник

Ответы (1)


arrow_upward
0
arrow_downward

  1. Найдите https://download.pytorch.org/whl/torch_stable.html для версия, которую вы хотите установить (версия torchvision, версия python, версия CUDA, ОС и т. д.)
  2. Добавьте зависимость URL к вашему файлу pyproject.toml. Например, у меня есть torchvision 1.8.0, работающий со следующими зависимостями:
[tool.poetry.dependencies]
python = "^3.8"
torch = {url = "https://download.pytorch.org/whl/cu102/torch-1.8.0-cp38-cp38-win_amd64.whl"}
torchvision = {url = "https://download.pytorch.org/whl/cu102/torchvision-0.9.0-cp38-cp38-win_amd64.whl"}
  1. В активированной виртуальной среде выполните poetry update torchvision, и все будет готово.
person Dris101    schedule 13.03.2021