如果通过设置gpu实现神经网络训练

这期内容当中小编将会给大家带来有关如果通过设置gpu实现神经网络训练，文章内容丰富且以专业的角度为大家分析和叙述，阅读完这篇文章希望大家可以有所收获。

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

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

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

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

上述就是小编为大家分享的如果通过设置gpu实现神经网络训练了，如果刚好有类似的疑惑，不妨参照上述分析进行理解。如果想知道更多相关知识，欢迎关注亿速云行业资讯频道。