基于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和GO语言实现的消息摘要算法示例
Mar 10 Python
python中的代码编码格式转换问题
Jun 10 Python
Python基于socket模块实现UDP通信功能示例
Apr 10 Python
对Pandas DataFrame缺失值的查找与填充示例讲解
Nov 06 Python
解决pycharm的Python console不能调试当前程序的问题
Jan 20 Python
对Python生成汉字字库文字,以及转换为文字图片的实例详解
Jan 29 Python
Pandas 解决dataframe的一列进行向下顺移问题
Dec 27 Python
python在不同条件下的输入与输出
Feb 13 Python
Pycharm如何运行.py文件的方法步骤
Mar 03 Python
Python3自定义http/https请求拦截mitmproxy脚本实例
May 11 Python
Python定时任务框架APScheduler原理及常用代码
Oct 05 Python
pycharm 的Structure界面设置操作
Feb 05 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
Apache环境下PHP利用HTTP缓存协议原理解析及应用分析
2010/02/16 PHP
php小经验:解析preg_match与preg_match_all 函数
2013/06/29 PHP
Firefox outerHTML实现代码
2009/06/04 Javascript
javascript类继承机制的原理分析
2009/09/12 Javascript
location.search在客户端获取Url参数的方法
2010/06/08 Javascript
JavaScript继承方式实例
2010/10/29 Javascript
JS 进度条效果实现代码整理
2011/05/21 Javascript
javascript学习基础笔记之DOM对象操作
2011/11/03 Javascript
利用javascript判断文件是否存在
2013/12/31 Javascript
JS遍历Json字符串中键值对先转成JSON对象再遍历
2014/08/15 Javascript
Javascript数据结构与算法之列表详解
2015/03/12 Javascript
js实现人民币大写金额形式转换
2016/04/27 Javascript
基于jQuery实现文字打印动态效果
2017/04/21 jQuery
vue axios用法教程详解
2017/07/23 Javascript
vue项目中v-model父子组件通信的实现详解
2017/12/10 Javascript
通过实例讲解JS如何防抖动
2019/06/15 Javascript
10种JavaScript最常见的错误(小结)
2019/06/21 Javascript
vue-property-decorator用法详解
2019/12/12 Javascript
40行代码把Vue3的响应式集成进React做状态管理
2020/05/20 Javascript
vue设置全局访问接口API地址操作
2020/08/14 Javascript
python实现的系统实用log类实例
2015/06/30 Python
Python中使用Queue和Condition进行线程同步的方法
2016/01/19 Python
Python找出list中最常出现元素的方法
2016/06/14 Python
pyhton列表转换为数组的实例
2018/04/04 Python
python实现黑客字幕雨效果
2018/06/21 Python
对python3 Serial 串口助手的接收读取数据方法详解
2019/06/12 Python
浅析pip安装第三方库及pycharm中导入第三方库的问题
2020/03/10 Python
Python基础教程之输入输出和运算符
2020/07/26 Python
CSS3中的Media Queries学习笔记
2016/05/23 HTML / CSS
澳大利亚运动鞋商店:Platypus Shoes
2019/09/27 全球购物
农行实习自我鉴定
2013/09/22 职场文书
通用求职信范文模板分享
2013/12/27 职场文书
网络工程专业自荐信范文
2014/03/16 职场文书
护理实习生带教计划
2015/01/16 职场文书
2015年重阳节主持词
2015/07/04 职场文书
Python基础详解之邮件处理
2021/04/28 Python