欢迎光临
我们一直在努力

玩转TensorFlow分布式(多GPU和多服务器)详解

本节介绍有关 TensorFlow 分布式的两个实际用例,分别是数据并行(将数据分布到多个 GPU 上)和多服务器分配。

玩转分布式TensorFlow:多个GPU和一个CPU

展示一个数据并行的例子,其中数据被切分到多个 GPU 上。

这个案例灵感来自 Neil Tenenholtz 的一篇很好的博客文章,网址为https://clindatsci.com/blog/2017/5/31/distributed-tensorflow。

具体做法

  1. 考虑在单个 GPU 上运行矩阵乘法的这段代码:



     

赞(0) 打赏
未经允许不得转载:hosteons中文网 » 玩转TensorFlow分布式(多GPU和多服务器)详解

评论 抢沙发

觉得文章有用就打赏一下文章作者

非常感谢你的打赏,我们将继续提供更多优质内容,让我们一起创建更加美好的网络世界!

支付宝扫一扫

微信扫一扫

登录

找回密码

注册