编程 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 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