基于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实现合成多张图片到PDF格式
在日常生活中，经常会遇到需要提交身份证正反面证明资料的情况，而且这些网站大部分只接受pdf格式，这时候我们就需要把身份证正反面两张图片合成为一个pdf文件。本文将为大家提供用Python实现这一要求的方法，需要的可以参考一下
2022-02-02
Django跨域请求CSRF的方法示例
这篇文章主要介绍了Django跨域请求CSRF的方法示例，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧
2018-11-11
解决ImportError: cannot import name ‘Imput
您遇到的ImportError: cannot import name ‘Imputer‘错误提示表明您尝试导入一个名为’Imputer’的模块或类,但是该模块或类无法找到,本文小编给大家介绍了如何解决这个问题,需要的朋友可以参考下
2023-10-10
基于pandas中expand的作用详解
今天小编就为大家分享一篇基于pandas中expand的作用详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2019-12-12
tensorflow 变长序列存储实例
今天小编就为大家分享一篇tensorflow 变长序列存储实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-01-01
使用Python对零售商品进行数据分析
这篇文章主要为大家介绍了使用Python对零售商品进行数据分析详解，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪
2022-05-05
Python函数装饰器常见使用方法实例详解
这篇文章主要介绍了Python函数装饰器常见使用方法,结合实例形式分析了Python函数装饰器的概念、原理、用法及相关操作注意事项,需要的朋友可以参考下
2019-03-03
详解python3 + Scrapy爬虫学习之创建项目
这篇文章主要介绍了python3 Scrapy爬虫创建项目，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2019-04-04
python 实现简单的计算器(gui界面)
这篇文章主要介绍了python 如何实现简单的计算器，帮助大家更好的理解和使用python，感兴趣的朋友可以了解下
2020-11-11
Python Datetime模块和Calendar模块用法实例分析
这篇文章主要介绍了Python Datetime模块和Calendar模块用法,结合实例形式分析了Python日期时间及日历相关的Datetime模块和Calendar模块原理、用法及操作注意事项,需要的朋友可以参考下
2019-04-04

基于Tensorflow使用CPU而不用GPU问题的解决

相关文章

最新评论

大家感兴趣的内容

最近更新的内容

常用在线小工具