基于Tensorflow使用CPU而不用GPU问题的解决
更新时间：2020年02月07日 08:38:19 作者：JR_Chan
今天小编就为大家分享一篇基于Tensorflow使用CPU而不用GPU问题的解决，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
之前的文章讲过用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 变量的合并几种方法，分享内容有list 合并和str 合并以及dict 合并的分析，下面具体方法介绍，需要的小伙伴可以参考一下2022-03-03
本文主要介绍了pyqt中解决国际化tr()函数不起作用的问题，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-02-02
CentOS6.9 Python环境配置（python2.7、pip、virtualenv）
这篇文章主要介绍了CentOS6.9 Python环境配置（python2.7、pip、virtualenv）方法，非常不错，具有一定的参考借鉴价值 ,需要的朋友可以参考下2019-05-05
这篇文章主要介绍了简单理解Python中基于生成器的状态机,来自于IBM官方技术文档,需要的朋友可以参考下2015-04-04
这篇文章主要介绍了Laravel+Dingo/Api 自定义响应的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2019-02-02
Python中Cookie模块（python3中为http.cookies）提供了一个类似字典的特殊对象SimpleCookie，其中存储并管理着称为Morsel的cookie值集合，这里介绍了python操作cookie的使用方法2014-03-03
人脸识别技术是一种通过计算机对人脸图像进行分析和处理,从而实现自动识别和辨认人脸的技术,随着计算机视觉和模式识别领域的快速发展,人脸识别技术取得了长足的进步,本文给大家介绍了基于Python实现人脸识别相似度对比功能,感兴趣的朋友可以参考下2024-01-01
这篇文章主要介绍了Python排序搜索基本算法之归并排序,简单描述了归并排序的特点,并结合实例形式分析了Python实现归并排序的具体操作技巧,需要的朋友可以参考下2017-12-12
urllib2是Python的一个获取URLs(Uniform Resource Locators)的组件。他以urlopen函数的形式提供了一个非常简单的接口，下面我们用实例讲解他的使用方法2014-11-11
这篇文章主要给大家介绍了关于Python中新式类与经典类的区别，文中通过示例代码介绍的非常详细，对大家学习或者使用Python具有一定的参考学习价值，需要的朋友们下面来一起学习学习吧2019-07-07
最新评论