编程 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查询Mysql时返回字典结构的代码

Jun 18 Python

分享python数据统计的一些小技巧

Jul 21 Python

Django卸载之后重新安装的方法

Mar 15 Python

python中使用正则表达式的连接符示例代码

Oct 10 Python

解决python 输出是省略号的问题

Apr 19 Python

pandas 条件搜索返回列表的方法

Oct 30 Python

Python 运行 shell 获取输出结果的实例

Jan 07 Python

python 动态调用函数实例解析

Oct 21 Python

Eclipse配置python默认头过程图解

Apr 26 Python

如何使用python写截屏小工具

Sep 29 Python

Python文件操作及内置函数flush原理解析

Oct 13 Python

Python爬虫过程解析之多线程获取小米应用商店数据

Nov 14 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中对2个数组相加的函数

2011/06/24 PHP

利用php绘制饼状图的实现代码

2013/06/07 PHP

php实现复制移动文件的方法

2015/07/29 PHP

PHP实现数据四舍五入的方法小结【4种方法】

2019/03/27 PHP

EXTJS记事本当CompositeField遇上RowEditor

2011/07/31 Javascript

javascript动态加载三

2012/08/22 Javascript

Struts2的s:radio标签使用及用jquery添加change事件

2013/04/08 Javascript

深入浅析JavaScript中with语句的理解

2016/05/12 Javascript

JS实现图片预览的两种方式

2017/06/27 Javascript

JS库之wow.js使用方法

2017/09/14 Javascript

webpack将js打包后的map文件详解

2018/02/22 Javascript

在vue中使用echarts图表实例代码详解

2018/10/22 Javascript

vue自定义键盘信息、监听数据变化的方法示例【基于vm.$watch】

2019/03/16 Javascript

js实现提交前对列表数据的增删改查

2020/01/16 Javascript

JavaScript实现表单验证功能

2020/12/09 Javascript

在Python中使用Mako模版库的简单教程

2015/04/08 Python

Python PyQt5标准对话框用法示例

2017/08/23 Python

django.db.utils.ProgrammingError: (1146, u“Table‘’ doesn’t exist”)问题的解决

2018/07/13 Python

Python之字典添加元素的几种方法

2020/09/30 Python

python 爬虫基本使用——统计杭电oj题目正确率并排序

2020/10/26 Python

pandas使用函数批量处理数据(map、apply、applymap)

2020/11/27 Python

快速解决pymongo操作mongodb的时区问题

2020/12/05 Python

使用CSS3的::selection改变选中文本颜色的方法

2015/09/29 HTML / CSS

canvas需要在标签里直接定义宽高

2014/12/17 HTML / CSS

HTML5 解决苹果手机不能自动播放音乐问题

2017/12/27 HTML / CSS

乌克兰香水和化妆品网站：Notino.ua

2018/03/26 全球购物

八年级美术教学反思

2014/02/02 职场文书

初中三年毕业生的自我评价分享

2014/02/14 职场文书

国际贸易毕业生自荐书

2014/06/22 职场文书

工资收入证明样本（5篇）

2014/09/16 职场文书

领导班子个人对照检查剖析材料

2014/09/29 职场文书

民主生活会批评与自我批评总结

2014/10/17 职场文书

小学语文教师年度考核个人总结

2015/02/05 职场文书

2015年幼儿园中班工作总结

2015/04/25 职场文书

如何用Laravel包含你自己的帮助函数

2021/05/27 PHP

win10搭建配置ftp服务器的方法

2022/08/05 Servers