用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案


Posted in Python onMarch 03, 2021

训练的时候当然用gpu,速度快呀。

我想用cpu版的tensorflow跑一下,结果报错,这个错误不太容易看懂。

用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案

大概意思是没找到一些节点。

后来发现原因,用gpu和cpu保存的pb模型不太一样,但是checkpoints文件是通用的。

使用tensorflow-cpu再把checkpoints文件重新转换一下pb文件就可以了。

完美解决!

补充:tensflow-gpu版的无数坑坑坑!(tf坑大总结)

自己的小本本,之前预装有的pycharm+win10+anaconda3+python3的环境

2019/3/24重新安装发现:目前CUDA10.1安装不了tensorflow1.13,把CUDA改为10.0即可(记得对应的cudann呀)

如果刚入坑,建议先用tensorflw学会先跑几个demo,等什么时候接受不了cpu这乌龟般的速度之时,就要开始尝试让gpu来跑了。

cpu跑tensorflow只需要在anaconda3下载。

安装cpu跑的tensorflow:

我的小本本目前已经是gpu版本,cpu版本下红圈里那个版本就好了!

用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案

安装好了后直接在python命令中输入

import tensorflow as tf

如果不报错说明调用成功。

查看目前tensorflow调用的是cpu还是gpu运行:

import tensorflow as tf
import numpy as np
a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[2, 3], name='a')
b = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0, 6.0], shape=[3, 2], name='b')
c = tf.matmul(a, b)
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))

然后把这段代码粘贴到编译器中运行,

看一下运行的结果中,调用运行的是什么

用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案

看给出的是gpu还是cpu就能判断目前运行的是哪一个了

安装gpu版本的tensorflow:

首先第一步要确定你的显卡是否为N卡,

然后上https://developer.nvidia.com/cuda-gpus去看看你的显卡是否被NVDIA允许跑机器学习

用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案

对于CUDA与cudann的安装:

需要到nvdia下载CUDA与cudann,这里最重要的是注意CUDA与cudann与tensorflow三者的搭配,

注意版本的搭配!!!

注意版本的搭配!!!

注意版本的搭配!!!

tensorflow在1.5版本以后才支持9.0以上的CUDA,所以如果CUDA版本过高,会造成找不到文件的错误。

在官网也可以看到CUDA搭配的cudann

用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案

在安装完了cudann时,需要把其三个文件复制到CUDA的目录下,并且添加3个新的path:

用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案

3个path,

用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案

当使用gpu版的tf时,就不再需要安装原来版本的tf,卸载了就好,安装tf-gpu版,

判断自己是否有安装tf包,对于pycharm用户,可以在setting那看看是否安装了tf-gpu

用gpu训练好的神经网络,用tensorflow-cpu跑出错的原因及解决方案

我使用的环境为:tf1.2+CUDA8.0+cudann5.1

当全部正确安装时

import tensorflow as tf 仍然出错

cudnn64_6.dll问题

关于导入TensorFlow找不到cudnn64_6.dll,其实下载的的是cudnn64_7.dll(版本不符合),把其修改过来就行了。

目录是在:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin下

对于不断尝试扔失败运行GPU版本,可以把tf先删除了重新下

import tensorflow as tf
print(tf.__version__)

查询tensorflow安装路径为:

print(tf.__path__)

成功用GPU运行但运行自己的代码仍然报错:

如果报错原因是这个

ResourceExhaustedError (see above for traceback): OOM when allocating tensor with shape[10000,28,28,32]

最后关于这个报错是因为GPU的显存不够,此时你可以看看你的代码,是训练集加载过多还是测试集加载过多,将它一次只加载一部分即可。

对于训练集banch_xs,banch_ys = mnist.train.next_batch(1000) 改为

banch_xs,banch_ys = mnist.train.next_batch(100)即可,

而测试集呢print(compute_accuracy(mnist.test.images[:5000], mnist.test.labels[:5000])) 改为

print(compute_accuracy(mnist.test.images, mnist.test.labels))即可

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

Python 相关文章推荐
Python socket.error: [Errno 98] Address already in use的原因和解决方法
Aug 25 Python
github配置使用指南
Nov 18 Python
Windows下安装python2.7及科学计算套装
Mar 05 Python
Python中super关键字用法实例分析
May 28 Python
Python 调用Java实例详解
Jun 02 Python
Python使用微信SDK实现的微信支付功能示例
Jun 30 Python
Python:Scrapy框架中Item Pipeline组件使用详解
Dec 27 Python
Python对excel文档的操作方法详解
Dec 10 Python
python 阶乘累加和的实例
Feb 01 Python
python实现统计文本中单词出现的频率详解
May 20 Python
django 实现celery动态设置周期任务执行时间
Nov 19 Python
Python 字典一个键对应多个值的方法
Sep 29 Python
神经网络训练采用gpu设置的方式
Mar 03 #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
You might like
php 多线程上下文中安全写文件实现代码
2009/12/28 PHP
解析php下载远程图片函数 可伪造来路
2013/06/25 PHP
PHP简单读取xml文件的方法示例
2017/04/20 PHP
PHP大文件分割上传 PHP分片上传
2017/08/28 PHP
php判断电子邮件是否正确方法
2018/12/04 PHP
ASP中Sub和Function的区别说明
2020/08/30 Javascript
jquery基础教程之deferred对象使用方法
2014/01/22 Javascript
jQuery+php实时获取及响应文本框输入内容的方法
2016/05/24 Javascript
PHP实现本地图片上传和验证功能
2017/02/27 Javascript
JS实现新建文件夹功能
2017/06/17 Javascript
详解bootstrap用dropdown-menu实现上下文菜单
2017/09/22 Javascript
jquery 输入框查找关键字并提亮颜色的实例代码
2018/01/23 jQuery
快速了解vue-cli 3.0 新特性
2018/02/28 Javascript
React diff算法的实现示例
2018/04/20 Javascript
解决vue中修改了数据但视图无法更新的情况
2018/08/27 Javascript
JS使用栈判断给定字符串是否是回文算法示例
2019/03/04 Javascript
jQuery实现B2B网站后台管理系统侧导航
2020/07/08 jQuery
vue动画—通过钩子函数实现半场动画操作
2020/08/09 Javascript
javascript实现移动端上传图片功能
2020/08/18 Javascript
python使用PyFetion来发送短信的例子
2014/04/22 Python
pandas数据框,统计某列数据对应的个数方法
2018/04/11 Python
python求质数的3种方法
2018/09/28 Python
解决python flask中config配置管理的问题
2019/07/26 Python
解决Keras 自定义层时遇到版本的问题
2020/06/16 Python
让IE支持CSS3的不完全兼容方案
2014/09/19 HTML / CSS
英国门销售网站:Green Tree Doors
2020/01/07 全球购物
汽车技术服务与营销专业在籍生自荐信
2013/09/28 职场文书
求职信的最佳写作思路
2014/02/01 职场文书
售前工程师职业生涯规划
2014/03/02 职场文书
爱心活动计划书
2014/04/26 职场文书
培训师岗位职责
2015/02/14 职场文书
图书借阅制度范本
2015/08/06 职场文书
高中数学教学反思范文
2016/02/18 职场文书
php 防护xss,PHP的防御XSS注入的终极解决方案
2021/04/01 PHP
解决SpringBoot跨域的三种方式
2021/06/26 Java/Android
Python 中的 copy()和deepcopy()
2021/11/07 Python