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

Netty 源码解析 ——— ChannelConfig 和 Attribute

源码解析:Spring源码解析笔记

Spring 源码解析之ViewResolver源码解析

Spring 源码解析之ViewResolver源码解析