基于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很早就引入了一种称为 f-string 的字符串格式化方法,它代表格式化字符串字面值,本文主要介绍了Python的f-string使用技巧,具有一定的参考价值,感兴趣的可以了解一下2024-01-01
在本篇文章里小编给大家整理的是一篇关于python中append函数用法讲解内容，有兴趣的朋友们可以学习下。2020-12-12
下面小编就为大家分享一篇python通过伪装头部数据抵抗反爬虫的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-05-05
Python socket网络编程TCP/IP服务器与客户端通信
这篇文章主要介绍了Python socket网络编程TCP/IP服务器与客户端通信的相关资料,这里对Scoket 进行详解并创建TCP服务器及TCP 客户端实例代码，需要的朋友可以参考下2017-01-01
这篇文章主要介绍了Python运算符重载用法,实例分析了Python运算符重载实现加法运算的技巧,需要的朋友可以参考下2015-05-05
下面小编就为大家带来一篇基于Python_脚本CGI、特点、应用、开发环境(详解)。小编觉得挺不错的，现在就分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-05-05
验证码是根据随机字符生成一幅图片，然后在图片中加入干扰象素，本文主要介绍了 TensorFlow2识别验证码，需要的朋友们下面随着小编来一起学习学习吧2021-06-06
Python爬虫是当下最火的一种获取数据的方式，当我们对一些小型网站进行爬取的时候往往没什么阻碍，而当我们爬取大型网站的时候经常会遇到禁止访问、封禁IP的情况，这也是我们触发反爬机制的体现，本文来带领大家了解几种简单高效的反爬对策2021-09-09
好的代码具有易理解、可扩展、易维护的特点，简直是人见人爱。本文就将介绍写好python代码的多个技巧2021-05-05
python线程池 ThreadPoolExecutor 的用法示例
这篇文章主要介绍了python线程池 ThreadPoolExecutor 的用法示例，帮助大家更好得理解和使用python线程池，感兴趣的朋友可以了解下2020-10-10
最新评论