基于Tensorflow使用CPU而不用GPU问题的解决


Posted in Python onFebruary 07, 2020

之前的文章讲过用Tensorflow的object detection api训练MobileNetV2-SSDLite,然后发现训练的时候没有利用到GPU,反而CPU占用率贼高(可能会有Could not dlopen library 'libcudart.so.10.0'之类的警告)。经调查应该是Tensorflow的GPU版本跟服务器所用的cuda及cudnn版本不匹配引起的。知道问题所在之后就好办了。

检查cuda和cudnn版本

 首先查看cuda版本:

cat /usr/local/cuda/version.txt

 以及cudnn版本:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

重新安装对应版本Tensorflow

 根据前面查看得到的cuda和cudnn版本,到Tensorflow官网查看对应的Tensorflow-GPU版本,然后用conda install tensorflow-gpu=[version]重新安装(把[version]换成对应的版本比如1.12)就OK了。

以上这篇基于Tensorflow使用CPU而不用GPU问题的解决就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
Python中for循环详解
Jan 17 Python
Python学习笔记之os模块使用总结
Nov 03 Python
Python3学习笔记之列表方法示例详解
Oct 06 Python
解决python 无法加载downsample模型的问题
Oct 25 Python
Python输出\u编码将其转换成中文的实例
Dec 15 Python
python+opencv打开摄像头,保存视频、拍照功能的实现方法
Jan 08 Python
pandas.cut具体使用总结
Jun 24 Python
python开启debug模式的方法
Jun 27 Python
Python基础之高级变量类型实例详解
Jan 03 Python
Python3爬虫里关于Splash负载均衡配置详解
Jul 10 Python
利于python脚本编写可视化nmap和masscan的方法
Dec 29 Python
一篇文章弄懂Python关键字、标识符和变量
Jul 15 Python
python实现ip地址的包含关系判断
Feb 07 #Python
tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现
Feb 06 #Python
Tensorflow中tf.ConfigProto()的用法详解
Feb 06 #Python
Python continue语句实例用法
Feb 06 #Python
Tensorflow限制CPU个数实例
Feb 06 #Python
如何解决tensorflow恢复模型的特定值时出错
Feb 06 #Python
python range实例用法分享
Feb 06 #Python
You might like
详解PHP中cookie和session的区别及cookie和session用法小结
2016/06/12 PHP
php json中文编码为null的解决办法
2016/12/14 PHP
php7函数,声明,返回值等新特性介绍
2018/05/25 PHP
js 加载时自动调整图片大小
2008/05/28 Javascript
Javascript学习笔记6 prototype的提出
2010/01/11 Javascript
js 省地市级联选择
2010/02/07 Javascript
JS限制上传图片大小不使用控件在本地实现
2012/12/19 Javascript
jQuery对html元素取值与赋值的方法
2013/11/20 Javascript
javascripit实现密码强度检测代码分享
2013/12/12 Javascript
基于Jquery+Ajax+Json实现分页显示附效果图
2014/07/30 Javascript
初识Node.js
2014/09/03 Javascript
BOOTSTRAP时间控件显示在模态框下面的bug修复
2015/02/05 Javascript
jQuery+HTML5实现手机摇一摇换衣特效
2015/06/05 Javascript
js操作cookie保存浏览记录的方法
2015/12/25 Javascript
使用jQuery.form.js/springmvc框架实现文件上传功能
2016/05/12 Javascript
jQuery插件Validation快速完成表单验证的方式
2016/07/28 Javascript
javascript数组遍历的方法实例分析
2016/09/13 Javascript
详解Javascript数据类型的转换规则
2016/12/12 Javascript
ES6新特性二:Iterator(遍历器)和for-of循环详解
2017/04/20 Javascript
基于Vue2.X的路由和钩子函数详解
2018/02/09 Javascript
2019 年编写现代 JavaScript 代码的5个小技巧(小结)
2019/01/15 Javascript
微信小程序保存图片到相册权限设置
2020/04/09 Javascript
微信小程序实现滑动操作代码
2020/04/23 Javascript
在Vue中使用Echarts可视化库的完整步骤记录
2020/11/18 Vue.js
python实现的一只从百度开始不断搜索的小爬虫
2013/08/13 Python
简单实现python爬虫功能
2015/12/31 Python
python使用KNN算法手写体识别
2018/02/01 Python
解决python2 绘图title,xlabel,ylabel出现中文乱码的问题
2019/01/29 Python
django序列化serializers过程解析
2019/12/14 Python
pymysql的简单封装代码实例
2020/01/08 Python
基于PyQT实现区分左键双击和单击
2020/05/19 Python
Dyson加拿大官方网站:购买戴森吸尘器,风扇,冷热器及配件
2016/10/26 全球购物
美体小铺奥地利官方网站:The Body Shop奥地利
2019/04/11 全球购物
Harman Audio官方商店:购买JBL、Harman Kardon、Infinity和AKG
2019/12/05 全球购物
党员转正意见怎么写
2015/06/03 职场文书
心理健康教育主题班会
2015/08/13 职场文书