码迷,mamicode.com
首页 > 其他好文 > 详细

BytePS源码解析

时间:2019-08-26 14:53:32      阅读:101      评论:0      收藏:0      [点我收藏+]

标签: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

 

BytePS源码解析

标签:ted   color   源码解析   源码   bsp   code   ice   pre   table   

原文地址:https://www.cnblogs.com/yangwenhuan/p/11412275.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!