Trouble installing PyTorch for OpenNMT

Question

I'm trying to train an openNMT model using the instructions here - https://pypi.org/project/OpenNMT-py/

However I run into a problem while installing (using python setup.py install)

It runs into an error -

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')

I tried manually installing manually - python -m pip install torch==1.4.0 (because the link earlier says 1.4.0 is the only one supported, but it doesn't work. (Says No Matching Distribution found for torch==1.4.0)

Answer 1

this worked for me on Windows 10, first install pytorch

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

then run the script

python -m pip install OpenNMT-py