标签:ted color 源码解析 源码 bsp code ice pre table
import byteps.torch as bps
# 初始化BytePS bps.init()
# GPU和本地进程编号绑定 torch.cuda.set_device(bps.local_rank())
参考链接
https://pytorch.org/docs/stable/distributed.html
标签:ted color 源码解析 源码 bsp code ice pre table
原文地址:https://www.cnblogs.com/yangwenhuan/p/11412275.html