圆月山庄资源网 Design By www.vgjia.com

在定义图结构之前不用加入gpu:0,只有在session中计算之前在加入,否则的话会提示不能使用gpu保存模型等问题。

 with
  tf.device(
 '/gpu:0'
  ):

补充:关于应用gpu训练神经网络的注意事项

对于GPU来说,一定要注意的是,要分别在两个GPU上,或者不同时的在一个GPU上运行train和evaluation的部分,否则限于GPU擅长迭代而不擅长逻辑的特性,会发生OOM(out of memory)

神经网络训练采用gpu设置的方式

以上为个人经验,希望能给大家一个参考,也希望大家多多支持。如有错误或未考虑完全的地方,望不吝赐教。

标签:
神经网络训练,gpu设置

圆月山庄资源网 Design By www.vgjia.com
广告合作:本站广告合作请联系QQ:858582 申请时备注:广告合作(否则不回)
免责声明:本站文章均来自网站采集或用户投稿,网站不提供任何软件下载或自行开发的软件! 如有用户或公司发现本站内容信息存在侵权行为,请邮件告知! 858582#qq.com
圆月山庄资源网 Design By www.vgjia.com