基于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 2.7.x 和 3.x 版本的重要区别小结
Nov 28 Python
python WindowsError的错误代码详解
Jul 23 Python
selenium python浏览器多窗口处理代码示例
Jan 15 Python
Tornado高并发处理方法实例代码
Jan 15 Python
Appium Python自动化测试之环境搭建的步骤
Jan 23 Python
pycharm new project变成灰色的解决方法
Jun 27 Python
简单了解python协程的相关知识
Aug 31 Python
Python中关于浮点数的冷知识
Sep 22 Python
python实现连续变量最优分箱详解--CART算法
Nov 22 Python
python中如何使用insert函数
Jan 09 Python
Python集成开发工具Pycharm的安装和使用详解
Mar 18 Python
python全栈开发语法总结
Nov 22 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
一个MYSQL操作类
2006/11/16 PHP
处理单名多值表单的详解
2013/06/08 PHP
zf框架的zend_cache缓存使用方法(zend框架)
2014/03/14 PHP
PHP二维数组排序简单实现方法
2016/02/14 PHP
Laravel中基于Artisan View扩展包创建及删除应用视图文件的方法
2016/10/08 PHP
javascript 带有滚动条的表格,标题固定,带排序功能.
2009/11/13 Javascript
javascript中创建对象的几种方法总结
2013/11/01 Javascript
js 剪切板应用clipboardData详细解析
2013/12/17 Javascript
javascript特殊日历控件分享
2016/03/07 Javascript
jQuery 全选 全部选 反选 实现代码
2016/08/17 Javascript
微信小程序之购物车功能
2020/09/23 Javascript
swiper移动端轮播插件(触碰图片之后停止轮播)
2017/12/28 Javascript
vue+element实现批量删除功能的示例
2018/02/28 Javascript
微信小程序用户信息encryptedData详解
2018/08/24 Javascript
js中的闭包实例展示
2018/11/01 Javascript
vue自定义指令实现方法详解
2019/02/11 Javascript
在Layui中实现开关按钮的效果实例
2019/09/29 Javascript
js判断非127开头的IP地址的实例代码
2020/01/05 Javascript
[01:33]DOTA2上海特级锦标赛 LIQUID战队完整宣传片
2016/03/16 DOTA
Python实现图像几何变换
2015/07/06 Python
python hook监听事件详解
2018/10/25 Python
使用Rasterio读取栅格数据的实例讲解
2019/11/26 Python
tensorflow实现tensor中满足某一条件的数值取出组成新的tensor
2020/01/04 Python
html5中JavaScript removeChild 删除所有节点
2014/05/16 HTML / CSS
HTML5 创建canvas元素示例代码
2014/06/04 HTML / CSS
Spartoo美国:欧洲排名第一的在线时装零售商
2019/12/12 全球购物
乌克兰品牌化妆品和香水在线商店:Bomond
2020/01/14 全球购物
Structs界面控制层技术
2013/10/11 面试题
学校后勤人员职责
2013/12/27 职场文书
房贷工资证明范本
2015/06/12 职场文书
文艺部部长竞选稿
2015/11/21 职场文书
蓝天保卫战收官在即 :15行业将开展环保分级评价
2019/07/19 职场文书
Winsows11性能如何? win11性能测评多核竟比Win10差了10%
2021/11/21 数码科技
使用 Apache 反向代理的设置技巧
2022/01/18 Servers
详解使用内网穿透工具Ngrok代理本地服务
2022/03/31 Servers
2022漫威和DC电影上映作品
2022/04/05 欧美动漫