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

  • 内容
  • 评论
  • 相关

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

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

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

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

具体做法

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



     

本文标题:玩转TensorFlow分布式(多GPU和多服务器)详解

本文地址:https://www.hosteonscn.com/4311.html

评论

0条评论

发表评论

邮箱地址不会被公开。 必填项已用*标注