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

软件发布|下载排行|最新软件

当前位置:首页IT学院IT技术

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

Destiny_Ren   2021-03-03 我要评论
想了解神经网络训练采用gpu设置的方式的相关内容吗,Destiny_Ren在本文为您仔细讲解神经网络训练gpu设置的相关知识和一些Code实例,欢迎阅读和指正,我们先划重点:神经网络训练,gpu设置,下面大家一起来学习吧。

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

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

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

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

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

Copyright 2022 版权所有 软件发布 访问手机版

声明:所有软件和文章来自软件开发商或者作者 如有异议 请与本站联系 联系我们