python-2.7 - 如何为 OpenNMT-py (pytorch) 提供多 GPU 支持？

Question

我使用 python-2.7 版本来运行支持 GPU 的 PyTorch。我使用此命令使用多 GPU 训练数据集。有人可以告诉我如何在 OpenNMT-py 中使用 PyTorch 修复此错误，或者有没有办法使用 python 2.7 为多 GPU 提供 pytorch 支持？

这是我尝试过的命令。

CUDA_VISIBLE_DEVICES=1,2 python train.py -data data/demo -save_model demo-model -world_size 2 -gpu_ranks 0 1

这是错误：

回溯（最后一次调用）：文件“train.py”，第 200 行，在 main(opt) 文件“train.py”，第 60 行，在 main mp = torch.multiprocessing.get_context('spawn') AttributeError: '模块'对象没有属性'get_context'

标签： python-2.7pytorchopennmt

Answer 1

也许您可以检查您的 torch 和 python 版本是否符合 openmt 要求。我记得他们的手电筒是 1.0 或 1.2（1.0 更好）。您必须降低最新版本的火炬。希望这会奏效