Tensorflow 多gpu训练
Posted 573177885qq
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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
以上是关于Tensorflow 多gpu训练的主要内容,如果未能解决你的问题,请参考以下文章
『TensorFlow』分布式训练_其二_多GPU并行demo分析(待续)