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

脚本专栏 2024/12/23 佚名

3 2 1

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

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

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

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

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

神经网络训练,gpu设置

标签：

免责声明：本站文章均来自网站采集或用户投稿，网站不提供任何软件下载或自行开发的软件！如有用户或公司发现本站内容信息存在侵权行为，请邮件告知！ 858582#qq.com

白云城资源网 Copyright www.dyhadc.com

评论“神经网络训练采用gpu设置的方式”

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

暂无“神经网络训练采用gpu设置的方式”评论...

2024年12月23日