为 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 是唯一受支持的，但它不起作用。(表示没有找到火炬的匹配分布==1.4.0(

Answer 1

这在 Windows 10 上对我有用，首先安装 pytorch

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

然后运行脚本

python -m pip install OpenNMT-py