脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Python - 神经网络训练采用gpu设置的方式

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

2021-09-13 00:16Destiny_Ren Python

这篇文章主要介绍了神经网络训练采用gpu设置的方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧

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

?
1
2
3
4
with
 tf.device(
'/gpu:0'
 ):

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

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

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

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

原文链接:https://blog.csdn.net/weixin_41863859/article/details/80222361

延伸 · 阅读

精彩推荐