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

Tensorflow 多gpu训练

时间:2019-08-31 19:37:15      阅读:82      评论:0      收藏:0      [点我收藏+]

标签:igp   训练   tps   使用   book   examples   git   force   multi   

Tensorflow可在训练时制定占用那几个gpu,但如果想真正的使用多gpu训练,则需要手动去实现。

不知道tf2会不会改善一下。

具体参考:https://wizardforcel.gitbooks.io/tensorflow-examples-aymericdamien/6.2_multigpu_cnn.html

                https://jhui.github.io/2017/03/07/TensorFlow-GPU/

                https://zhuanlan.zhihu.com/p/75195049

 

Tensorflow 多gpu训练

标签:igp   训练   tps   使用   book   examples   git   force   multi   

原文地址:https://www.cnblogs.com/573177885qq/p/11439942.html

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