Проблемы с установкой PyTorch для OpenNMT

Question

Я пытаюсь обучить модель openNMT, используя приведенные здесь инструкции - https://pypi.org/project/OpenNMT-py/

Однако я столкнулся с проблемой при установке (используя python setup.py install)

Произошла ошибка -

Processing dependencies for OpenNMT-py==1.1.1
Searching for torch>=1.4.0
Reading https://pypi.org/simple/torch/
No local packages or working download links found for torch>=1.4.0
error: Could not find suitable distribution for Requirement.parse('torch>=1.4.0')

Я пробовал вручную установить - python -m pip install torch==1.4.0 (потому что ссылка ранее говорит, что 1.4.0 - единственный поддерживаемый, но он не работает (говорит) Соответствующее распределение не найдено для torch == 1.4.0)

Answer 1

это сработало для меня на Windows 10, сначала установите pytorch

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

затем запустите скрипт

python -m pip install OpenNMT-py