基于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 JSON编解码方式原理详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-01-01
本文主要介绍了Django之第三方平台QQ授权登录的实现，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2023-05-05
Python中的collections集合与typing数据类型模块
这篇文章介绍了Python中的collections集合与typing数据类型模块，文中通过示例代码介绍的非常详细。对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-05-05
师兄采集的视网膜图像放在一个文件夹下，由于分类十分混乱，因此出现了一个文件多次出现的情况，因此让我写了个脚本2013-04-04
这篇文章主要介绍了Python实现大数据收集至excel的思路，本文通过完整代码给大家介绍的非常详细，具有一定的参考借鉴价值,需要的朋友可以参考下2020-01-01
这篇文章主要介绍了python操作mysql、excel、pdf的示例，帮助大家更好的理解和学习使用python，感兴趣的朋友可以了解下2021-03-03
这篇文章主要介绍了pandas 使用merge实现百倍加速的操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-04-04
关于Torch torchvision Python版本对应关系说明
这篇文章主要介绍了关于Torch torchvision Python版本对应关系说明，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-05-05
在前面的文章中，我有分享到 vue+drf+第三方滑动验证码接入的实现，本文将要分享的是基于 python 实现原生的登录验证码,需要的朋友可以参考一下2021-10-10
这篇文章主要介绍了python实现文件名批量替换和内容替换，第一个例子可以指定文件类型,需要的朋友可以参考下2014-03-03
最新评论