编程 Python

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

Posted in Python onMarch 03, 2021

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

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

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

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

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

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

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

- Author -

Destiny_Ren

声明：登载此文出于传递更多信息之目的，并不意味着赞同其观点或证实其描述。

Python 相关文章推荐

从零学Python之入门（四）运算

May 27 Python

详解Python中expandtabs()方法的使用

May 18 Python

python实现自主查询实时天气

Jun 22 Python

使用Selenium破解新浪微博的四宫格验证码

Oct 19 Python

使用 Python 遍历目录树的方法

Feb 29 Python

Python使用requests xpath 并开启多线程爬取西刺代理ip实例

Mar 06 Python

Python如何用wx模块创建文本编辑器

Jun 07 Python

pandas处理csv文件的方法步骤

Oct 16 Python

Flask中jinja2的继承实现方法及实例

Mar 03 Python

Python办公自动化之Excel（中）

May 24 Python

Python中基础数据类型 set集合知识点总结

Aug 02 Python

Python matplotlib多个子图绘制整合

Apr 13 Python

解决TensorFlow训练模型及保存数量限制的问题

Mar 03 #Python

使paramiko库执行命令时在给定的时间强制退出功能的实现

Mar 03 #Python

python上下文管理的使用场景实例讲解

Mar 03 #Python

pip install命令安装扩展库整理

Mar 02 #Python

python实现杨辉三角的几种方法代码实例

Mar 02 #Python

python爬虫破解字体加密案例详解

Mar 02 #Python

python包的导入方式总结

Mar 02 #Python

You might like

深入了解php4(2)--重访过去

2006/10/09 PHP

PHP实现MySQL更新记录的代码

2008/06/07 PHP

取得一定长度的内容，处理中文

2006/12/20 Javascript

$.ajax json数据传递方法

2008/11/19 Javascript

Javascript和Ajax中文乱码吐血版解决方案

2009/12/21 Javascript

JS定义回车事件(实现代码)

2013/07/08 Javascript

javascript中typeof的使用示例

2013/12/19 Javascript

js Object2String方便查看js对象内容

2014/11/24 Javascript

javascript获得当前的信息的一些常用命令

2015/02/25 Javascript

js实现简单的可切换选项卡效果

2015/04/10 Javascript

使用Promise解决多层异步调用的简单学习心得

2016/05/17 Javascript

全面解析Bootstrap中scrollspy（滚动监听）的使用方法

2016/06/06 Javascript

Bootstrap BootstrapDialog使用详解

2017/02/17 Javascript

快速使用node.js进行web开发详解

2017/04/26 Javascript

DataTables添加额外的查询参数和删除columns等无用参数实例

2017/07/04 Javascript

记一次用vue做的活动页的方法步骤

2019/04/11 Javascript

Vue使用axios出现options请求方法

2019/05/30 Javascript

element-ui 中使用upload多文件上传只请求一次接口

2019/07/19 Javascript

[05:16]《大圣！大圣》——DOTA2新英雄齐天大圣配音李世宏老师专访

2016/12/13 DOTA

Python库urllib与urllib2主要区别分析

2014/07/13 Python

Python自动调用IE打开某个网站的方法

2015/06/03 Python

详解Django之auth模块（用户认证）

2018/04/17 Python

python生成1行四列全2矩阵的方法

2018/08/04 Python

Python中分支语句与循环语句实例详解

2018/09/13 Python

Pandas之DataFrame对象的列和索引之间的转化

2019/06/25 Python

python unichr函数知识点总结

2020/12/16 Python

Pycharm 设置默认解释器路径和编码格式的操作

2021/02/05 Python

CSS3 background-image颜色渐变的实现代码

2018/09/13 HTML / CSS

现代绅士日常奢侈品：Todd Snyder

2019/12/13 全球购物

Fnac西班牙官网：法国文化和电子产品零售商

2021/03/14 全球购物

什么是组件架构

2016/05/15 面试题

法律专业应届生自荐信范文

2014/01/06 职场文书

文化宣传方案

2014/03/13 职场文书

2015新年寄语大全

2014/12/08 职场文书

高校自主招生教师推荐信

2015/03/23 职场文书

2016年度师德标兵先进事迹材料

2016/02/26 职场文书