Tensorflow 多gpu训练

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

 

posted @ 2019-08-31 18:00  牧马人夏峥  阅读(332)  评论(0编辑  收藏  举报