神经网络训练采用gpu设置的方式
更新时间:2021年03月03日 08:45:16 作者:Destiny_Ren
这篇文章主要介绍了神经网络训练采用gpu设置的方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
在定义图结构之前不用加入gpu:0,只有在session中计算之前在加入,否则的话会提示不能使用gpu保存模型等问题。
with tf.device( '/gpu:0' ):
补充:关于应用gpu训练神经网络的注意事项
对于GPU来说,一定要注意的是,要分别在两个GPU上,或者不同时的在一个GPU上运行train和evaluation的部分,否则限于GPU擅长迭代而不擅长逻辑的特性,会发生OOM(out of memory)
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。如有错误或未考虑完全的地方,望不吝赐教。
相关文章
Tensorflow 实现将图像与标签数据转化为tfRecord文件
今天小编就为大家分享一篇Tensorflow 实现将图像与标签数据转化为tfRecord文件,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧2020-02-02详解python中[-1]、[:-1]、[::-1]、[n::-1]使用方法
这篇文章主要介绍了详解python中[-1]、[:-1]、[::-1]、[n::-1]使用方法,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2021-04-04
最新评论