在 GPU 中支持更多等级pytorch/openmpi

问题描述

目前我使用 Pytorch 和 OpenMPI 后端进行分布式训练。我有 2 个等级和 2 个 GPU,其中每个等级分配给一个 GPU。

我想再次使用两个 GPU 将等级数增加到 10。 有没有办法在单个 GPU 中拥有更多等级(例如,GPU 0 上有 0-4 个等级,GPU 1 上有 5-9 个等级)?

解决方法

暂无找到可以解决该程序问题的有效方法,小编努力寻找整理中!

如果你已经找到好的解决方法,欢迎将解决方案带上本链接一起发送给小编。

小编邮箱:dio#foxmail.com (将#修改为@)