SC21:OpenGraphGym-MG 描述: OpenGraphGym-MG是一款使用强化学习解决MultiGPU系统上大型图优化问题的软件 要求: Python 3.6.10 NetworkX 2.3 脾气暴躁的1.17.4 科学1.3.1 PyTorch 1.3.1 node2Vec 0.4.1 mpi4py 3.0.3 在OLCF Summit上,我们使用以下命令来满足要求: module load ibm-wml-ce # python 3.6.10, networkx 2.3, numpy 1.17.4, scipy 1.3.1, torch 1.3.1 pip install node2vec # 0.4.1 pip install mpi4py # 3.0.3 代码结构: ├── convergence_evaluation