安装tensorflow的时候,通常会遇到tf.test.is_gpu_available() 返回的false的问题,这时候的GPU是不可用的,并且会提示lib cudnn.so.8不存在,这通常是因为以下情况:

1、环境变量LD_LIBRARY_PATH 设置错误,

这个变量要指向lib.cudnn.so.8这个文件的父目录;

解决方法:修改环境变量。

vi ~/.bashrc

然后修改环境变量。

只要增加红框的代码;

2 、没有安装cudnn;

没有安装cudnn的话,要去cuDNN Archive | NVIDIA Developer 下载对应的版本,和cuda版本匹配,否则安装会出错,如果是linux的话,下载的deb文件(deb文件要匹配自己的版本)通过以下命令执行:

sudo dpkg -i libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb 

安装后就会有libcudnn.so.8这个文件了,我的路径在/usr/lib/x86_64-linux-gnu下,如果不在这,可以通过deb文件下的data.tar.xz下看一下libcudnn.so.8被安装到哪儿了。

 

Logo

为开发者提供学习成长、分享交流、生态实践、资源工具等服务,帮助开发者快速成长。

更多推荐