详解tensorflow2.x版本无法调用gpu的一种解决方法


Posted in Python onMay 25, 2020

最近学校给了一个服务器账号用来训练神经网络使用,服务器本身配置是十路titan V,然后在上面装了tensorflow2.2,对应的python版本是3.6.2,装好之后用tf.test.is_gpu_available()查看是否能调用gpu,结果返回结果是false,具体如下:

详解tensorflow2.x版本无法调用gpu的一种解决方法

这里tensorflow应该是检测出了gpu,但是因为某些库无法打开而导致tensorflow无法调用,返回了false,详细查看错误信息可以看到一行:

详解tensorflow2.x版本无法调用gpu的一种解决方法

可以看到上面几个文件都顺利打开了,但是最后一个libcudnn.so.7文件显示无法打开,not such file or directory。怀疑是cuda有问题,服务器本身是装的cuda10.1,跟tensorflow2.2应该是吻合的,但是一直无法调用,所以一开始我想重新安装cuda,覆盖掉服务器本来的cuda,下好安装包之后,因为我不是管理员,没有root权限,所以总是失败。但是安装过程中了解到系统的cuda安装目录,位于/usr/local/cuda下面,这个libcudnn.so.7应该是一个库文件,那应该放在cuda的安装目录下面,具体地,在/usr/local/cuda/lib64下面,之前在我的windows本地机器安装cuda时还要下载cudnn7.x,然后把文件拷贝到cuda对应的目录下面,我怀疑lib64目录下面的这个libcudnn.so.7文件有问题,因为在linux版本的cudnn中是可以看到libcudnn.so.7这个文件的。
于是,打开lib64目录,查找是否有libcudnn.so.7这个文件,结果是没有找到这个文件,这就很奇怪了,cuda10.1目录下面竟然没有cudnn的文件,我也没有权限修改/usr/local,因此想到既然是少了这个文件,那是不是把对应的文件加载在别的目录下,引导tensorflow去另一个我可以操作的目录下找这个libcudnn.so.7文件,就可以解决问题呢?我带着试验性的心态wget下载cudnn,结果在命令行下无法下载cudnn,原因是cudnn下载需要登录,而在命令行下就没办法下载。于是在本地机器上下载了linux版的cudnn,然后用scp命令把这个tar.gz文件发到服务器上,解压出来可以看到~/cudnn/cuda/lib64下面有libcudnn.so.7。接下来就是添加环境变量,让tensorflow不仅在/usr/local/cuda/lib64下找文件,还可以在我这个目录下找,添加命令:

export PATH=$PATH:/usr/local/cuda-10.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.1/lib64

这几行命令添加系统cuda的库
然后添加:

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/users/cudnn/cuda/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/home/users/cudnn/cuda/lib64

这里的/home/users/cudnn/cuda/lib64视自己的libcudnn.so.7文件目录而定,最后是更新:

source /etc/profile

如果tensorflow是按照在anaconda虚拟环境下的,在执行这几条命令会自动退出虚拟环境。记得重新进入:

source activate 环境名

这时重新进入python,导入tensorflow,然后运行tf.test.is_gpu_available(),可以看到:

详解tensorflow2.x版本无法调用gpu的一种解决方法

这时已经显示成功打开了libcudnn.so.7文件,说明tf根据我提供的目录成功找到了这个文件,在打开了这些库文件后,最下面也返回了True:

详解tensorflow2.x版本无法调用gpu的一种解决方法

还可以用另一个方法tf.config.list_physical_devices(‘GPU')查看当前可用gpu:

详解tensorflow2.x版本无法调用gpu的一种解决方法

十块gpu都显示出来了。
注意,添加这几条命令只在当前连接上有用,如果断开了服务器连接再重新连接时,需要重新输入这些命令。

这个方法只是作为一个参考,碰巧是在tensorflow2.2上这个libcudnn.so.7文件打不开,于是尝试性地试了一下,结果成功了。其他机器上如果出现同类问题,采用这个方法不一定能解决,只是提供一个思路。在tensorflow2.1上,也同样出现gpu无法调用的问题,但打印的错误信息不仅有libcudnn.so.7文件无法打开,还有其他几个文件也打不开,这些文件基本都是lib开头的,可以查看这些文件是否在cuda的lib64目录下,如果找得到这些文件,那有可能是环境变量设错了,可以试试上面那些命令:

export PATH=$PATH:/usr/local/cuda-10.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.1/lib64

如果找不到,那可以试试跟我一样先把这些文件下载到机器上面,给系统添加环境变量到对应这些文件的目录,引导tf去找。当然,这只是我一种猜测,tensorflow2.1和2.2用的应该都是cuda10.1,但是我不清楚为什么2.2只有一个文件无法打开,而2.1就有好几个文件打不开,而在1.9版本上,由于1.9似乎用的不是10.1版本的cuda,其错误原因更多,这里要区分开。

到此这篇关于详解tensorflow2.x版本无法调用gpu的一种解决方法的文章就介绍到这了,更多相关tensorflow2.x无法调用gpu内容请搜索三水点靠木以前的文章或继续浏览下面的相关文章希望大家以后多多支持三水点靠木!

Python 相关文章推荐
打印出python 当前全局变量和入口参数的所有属性
Jul 01 Python
python使用新浪微博api上传图片到微博示例
Jan 10 Python
python进阶教程之动态类型详解
Aug 30 Python
Python中使用hashlib模块处理算法的教程
Apr 28 Python
Using Django with GAE Python 后台抓取多个网站的页面全文
Feb 17 Python
Python之dict(或对象)与json之间的互相转化实例
Jun 05 Python
图文详解python安装Scrapy框架步骤
May 20 Python
Python:二维列表下标互换方式(矩阵转置)
Dec 02 Python
Python如何使用argparse模块处理命令行参数
Dec 11 Python
python错误调试及单元文档测试过程解析
Dec 19 Python
浅析Python中字符串的intern机制
Oct 03 Python
Selenium结合BeautifulSoup4编写简单的python爬虫
Nov 06 Python
keras模型保存为tensorflow的二进制模型方式
May 25 #Python
keras 如何保存最佳的训练模型
May 25 #Python
keras处理欠拟合和过拟合的实例讲解
May 25 #Python
python如何调用字典的key
May 25 #Python
如何使用python的ctypes调用医保中心的dll动态库下载医保中心的账单
May 24 #Python
Python+PyQt5实现灭霸响指功能
May 25 #Python
PyQt5实现仿QQ贴边隐藏功能的实例代码
May 24 #Python
You might like
PHP截取指定图片大小的方法
2014/12/10 PHP
php获取数组元素中头一个数组元素值的实现方法
2014/12/20 PHP
php获取远程文件内容的函数
2015/11/02 PHP
php利用ffmpeg提取视频中音频与视频画面的方法详解
2017/06/07 PHP
PHP使用redis消息队列发布微博的方法示例
2017/06/22 PHP
javascript完美拖拽的实现方法
2013/09/29 Javascript
js实现图片放大和拖拽特效代码分享
2015/09/05 Javascript
JS实现列表页面隔行变色效果
2017/03/25 Javascript
Vue基于NUXT的SSR详解
2017/10/24 Javascript
inquirer.js一个用户与命令行交互的工具详解
2019/05/18 Javascript
微信小程序自定义头部导航栏和导航栏背景图片 navigationStyle问题
2019/07/26 Javascript
vue中axios防止多次触发终止多次请求的示例代码(防抖)
2020/02/16 Javascript
JavaScript中的this妙用实例分析
2020/05/09 Javascript
vue 项目软键盘回车触发搜索事件
2020/09/09 Javascript
用vite搭建vue3应用的实现方法
2021/02/22 Vue.js
[02:45]DOTA2英雄基础教程 伐木机
2013/12/23 DOTA
[03:02]2020完美世界城市挑战赛(秋季赛)总决赛回顾
2021/03/11 DOTA
Python sklearn KFold 生成交叉验证数据集的方法
2018/12/11 Python
python读取txt文件并取其某一列数据的示例
2019/02/19 Python
python实现鸢尾花三种聚类算法(K-means,AGNES,DBScan)
2019/06/27 Python
在Pytorch中使用样本权重(sample_weight)的正确方法
2019/08/17 Python
python Kmeans算法原理深入解析
2019/08/23 Python
Python Django中间件,中间件函数,全局异常处理操作示例
2019/11/08 Python
sklearn线性逻辑回归和非线性逻辑回归的实现
2020/06/09 Python
Python Celery异步任务队列使用方法解析
2020/08/10 Python
html5的canvas实现3d雪花飘舞效果
2013/12/27 HTML / CSS
意大利奢侈品购物网站:Giglio
2018/01/05 全球购物
应届生财务管理求职信
2013/11/06 职场文书
会计辞职信范文
2014/01/15 职场文书
大学生学习2014年全国两会心得体会
2014/03/12 职场文书
对照四风自我剖析材料
2014/10/07 职场文书
2015年教师新年寄语
2014/12/08 职场文书
自我检讨书范文
2015/01/28 职场文书
学生逃课万能检讨书2000字
2015/02/17 职场文书
2015年入党积极分子培养考察意见
2015/08/12 职场文书
MySQL数据库必备之条件查询语句
2021/10/15 MySQL