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使用horovod进行多gpu训练

tensorflow能不能gpu cpu 一起训练

深度学习TensorFlow如何使用多GPU并行模式?

『TensorFlow』分布式训练_其二_多GPU并行demo分析(待续)

tensorflow 单机多GPU训练时间比单卡更慢/没有很大时间上提升

TensorFlow分布式部署单机多卡