神经网络训练采用gpu设置的方式
作者:Destiny_Ren 时间:2021-04-24 16:43:25
在定义图结构之前不用加入gpu:0,只有在session中计算之前在加入,否则的话会提示不能使用gpu保存模型等问题。
with
tf.device(
'/gpu:0'
):
补充:关于应用gpu训练神经网络的注意事项
对于GPU来说,一定要注意的是,要分别在两个GPU上,或者不同时的在一个GPU上运行train和evaluation的部分,否则限于GPU擅长迭代而不擅长逻辑的特性,会发生OOM(out of memory)
以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。如有错误或未考虑完全的地方,望不吝赐教。
来源:https://blog.csdn.net/weixin_41863859/article/details/80222361
标签:神经网络,训练,gpu
![](/images/zang.png)
![](/images/jiucuo.png)
猜你喜欢
python web.py开发httpserver解决跨域问题实例解析
2021-02-21 15:44:35
网站中文字的视觉设计
2008-04-16 13:35:00
![](https://img.aspxhome.com/file/UploadPic/20084/16/2008416134135478s.jpg)
sqlserver通用的删除服务器上的所有相同后缀的临时表
2012-06-06 20:07:34
对numpy数据写入文件的方法讲解
2022-01-24 21:09:31
Python 解析库json及jsonpath pickle的实现
2021-09-10 00:27:14
![](https://img.aspxhome.com/file/2023/5/87925_0s.jpg)
从Vista地址栏到网站导航菜单
2009-09-08 12:35:00
![](https://img.aspxhome.com/file/UploadPic/20099/8/nav-62s.jpg)
使用python模块plotdigitizer抠取论文图片中的数据实例详解
2023-07-14 01:50:52
![](https://img.aspxhome.com/file/2023/9/127099_0s.png)
python获取磁盘号下盘符步骤详解
2022-10-05 08:32:32
![](https://img.aspxhome.com/file/2023/1/76981_0s.png)
asp如何用JMail POP3接收电子邮件?
2010-06-13 13:09:00
读写json中文ASCII乱码问题的解决方法
2023-02-13 02:32:25
使用SQL Server 2008中对象相关性
2008-12-12 06:36:00
jQuery入门之–属性(一)
2009-07-12 15:29:00
asp如何创建一个Exchange用户?
2009-11-14 20:52:00
分享一些可视信息设计资源
2009-10-06 15:19:00
![](https://img.aspxhome.com/file/UploadPic/200910/6/ui-info-share-banner-41s.jpg)
最新版 Windows10上安装Python 3.8.5的步骤详解
2021-12-31 00:50:29
![](https://img.aspxhome.com/file/2023/1/118331_0s.png)
解决iframe的frameborder在chrome/ff/ie下的差异
2023-08-09 00:23:24
KnockoutJS 3.X API 第四章之表单submit、enable、disable绑定
2023-08-13 03:12:44
![](https://img.aspxhome.com/file/2023/7/56017_0s.png)
python 配置uwsgi 启动Django框架的详细教程
2022-10-29 06:11:15
ASP实现网站智能分词搜索
2007-10-18 13:50:00
Python使用py2exe打包程序介绍
2023-11-12 21:29:55