安装用于OpenNMT的PyTorch时出现问题

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