python - 为 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