BytePS源码解析
Posted yangwenhuan
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了BytePS源码解析相关的知识,希望对你有一定的参考价值。
import byteps.torch as bps
# 初始化BytePS bps.init()
# GPU和本地进程编号绑定 torch.cuda.set_device(bps.local_rank())
参考链接
https://pytorch.org/docs/stable/distributed.html
以上是关于BytePS源码解析的主要内容,如果未能解决你的问题,请参考以下文章
Spring 源码解析之HandlerAdapter源码解析
spring系统架构源码解析AutowireCandidateResolver