神经网络训练采用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设置的方式

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

Python 相关文章推荐
全面解读Python Web开发框架Django
Jun 30 Python
python操作字典类型的常用方法(推荐)
May 16 Python
python代码 if not x: 和 if x is not None: 和 if not x is None:使用介绍
Sep 21 Python
详解python中的文件与目录操作
Jul 11 Python
scrapy爬虫实例分享
Dec 28 Python
Python3.4 tkinter,PIL图片转换
Jun 21 Python
python设置值及NaN值处理方法
Jul 03 Python
python for循环输入一个矩阵的实例
Nov 14 Python
用python标准库difflib比较两份文件的异同详解
Nov 16 Python
Python使用pandas和xlsxwriter读写xlsx文件的方法示例
Apr 09 Python
python地震数据可视化详解
Jun 18 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
php addslashes及其他清除空格的方法是不安全的
2012/01/25 PHP
PHP基于自定义类随机生成姓名的方法示例
2017/08/05 PHP
PHP 传输会话curl函数的实例详解
2017/09/12 PHP
PHP call_user_func和call_user_func_array函数的简单理解与应用分析
2019/11/25 PHP
javascript实现轮显新闻标题链接
2007/08/13 Javascript
使用Java实现简单的server/client回显功能的方法介绍
2013/05/03 Javascript
简单实用jquery版三级联动select示例
2013/07/04 Javascript
node.js中的fs.appendFile方法使用说明
2014/12/17 Javascript
B/S(Web)实时通讯解决方案分享
2017/04/06 Javascript
用 Vue.js 递归组件实现可折叠的树形菜单(demo)
2017/12/25 Javascript
JS实现的文字间歇循环滚动效果完整示例
2018/02/13 Javascript
layui获取多选框中的值方法
2018/08/15 Javascript
快速搭建Node.js(Express)用户注册、登录以及授权的方法
2019/05/09 Javascript
vue中echarts的用法及与elementui-select的协同绑定操作
2020/11/17 Vue.js
python使用pil进行图像处理(等比例压缩、裁剪)实例代码
2017/12/11 Python
Python实现购物车购物小程序
2018/04/18 Python
pyqt5中QThread在使用时出现重复emit的实例
2019/06/21 Python
python全栈知识点总结
2019/07/01 Python
Python爬虫图片懒加载技术 selenium和PhantomJS解析
2019/09/18 Python
Python实现点云投影到平面显示
2020/01/18 Python
利用python中集合的唯一性实现去重
2020/02/11 Python
利用python实现汉诺塔游戏
2021/03/01 Python
阿里巴巴英国:Alibaba英国
2019/12/11 全球购物
南京软件公司的.net程序员笔试题
2014/08/31 面试题
电子商务个人自荐信
2013/12/12 职场文书
电信营业员自我评价分享
2014/01/17 职场文书
省文明单位申报材料
2014/05/08 职场文书
花坛标语大全
2014/06/30 职场文书
党的群众路线教育实践活动个人剖析材料
2014/10/07 职场文书
教师党员批评与自我批评发言稿
2014/10/15 职场文书
2014年宣传部工作总结
2014/11/12 职场文书
党员先进事迹材料
2014/12/19 职场文书
酒店办公室主任岗位职责
2015/04/01 职场文书
2015年小班保育员工作总结
2015/05/27 职场文书
2016学校先进党组织事迹材料
2016/02/29 职场文书
MySQL配置主从服务器(一主多从)
2021/08/07 MySQL