如何将多GPU支持带到OpenNMT-py (pytorch)？

Question

我使用python-2.7版本来运行PyTorch并支持GPU。我使用这个命令使用多GPU来训练数据集。请有人告诉我，如何用OpenNMT中的PyTorch修复这个错误，或者是否有一种方法可以使用python2.7来支持多GPU？

这是我试过的命令。

CUDA_VISIBLE_DEVICES=1,2 python train.py -data data/演示-save_model演示-模型-world_size 2 -gpu_ranks 0 1

这是一个错误：

追溯(最近一次调用)：文件"train.py"，第200行，主(选择)文件“train.py”中的第60行，主mp =torch.multiprocessing.get_context(‘产卵’) AttributeError：‘模块’对象没有属性'get_context‘

Answer 1

也许您可以检查您的火炬和python版本是否符合openmt要求。我记得他们的火炬是1.0或1.2 (1.0更好)。你必须降低你的最新版本的火炬。希望那能起作用