企石镇做网站,怎么开通微信公众号,一款完整的app开发流程介绍,哈尔滨房地产型网站建设最近学校给了一个服务器账号用来训练神经网络使用#xff0c;服务器本身配置是十路titan V#xff0c;然后在上面装了tensorflow2.2#xff0c;对应的python版本是3.6.2#xff0c;装好之后用tf.test.is_gpu_available()查看是否能调用gpu#xff0c;结果返回结果是false服务器本身配置是十路titan V然后在上面装了tensorflow2.2对应的python版本是3.6.2装好之后用tf.test.is_gpu_available()查看是否能调用gpu结果返回结果是false具体如下这里tensorflow应该是检测出了gpu但是因为某些库无法打开而导致tensorflow无法调用返回了false详细查看错误信息可以看到一行可以看到上面几个文件都顺利打开了但是最后一个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/binexport LD_LIBRARY_PATH$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64export LIBRARY_PATH$LIBRARY_PATH:/usr/local/cuda-10.1/lib64这几行命令添加系统cuda的库然后添加export LD_LIBRARY_PATH$LD_LIBRARY_PATH:/home/users/cudnn/cuda/lib64export 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()可以看到这时已经显示成功打开了libcudnn.so.7文件说明tf根据我提供的目录成功找到了这个文件在打开了这些库文件后最下面也返回了True还可以用另一个方法tf.config.list_physical_devices(‘GPU)查看当前可用gpu十块gpu都显示出来了。注意添加这几条命令只在当前连接上有用如果断开了服务器连接再重新连接时需要重新输入这些命令。这个方法只是作为一个参考碰巧是在tensorflow2.2上这个libcudnn.so.7文件打不开于是尝试性地试了一下结果成功了。其他机器上如果出现同类问题采用这个方法不一定能解决只是提供一个思路。在tensorflow2.1上也同样出现gpu无法调用的问题但打印的错误信息不仅有libcudnn.so.7文件无法打开还有其他几个文件也打不开这些文件基本都是lib开头的可以查看这些文件是否在cuda的lib64目录下如果找得到这些文件那有可能是环境变量设错了可以试试上面那些命令export PATH$PATH:/usr/local/cuda-10.1/binexport LD_LIBRARY_PATH$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64export 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内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家
