非root 切换 cudnn

在linux系统上使用tensorflow的时候经常会碰到cudnn 版本不匹配的问题

这里介绍一种简单易用的方法切换cudnn供参考使用

 

首先去官网或其他可以下载的地方，下载指定版本的cudnn 解压包

举例：cudnn-9.0-linux-x64-v7.6.1.34.gtz 是cudnn7.6 版本的，而root权限下是cudnn7.3版

在使用tensorflow2.0时，需要使用cudnn7.6版

那么下载cudnn-9.0-linux-x64-v7.6.1.34.gtz 这个文件并解压缩文件能够得到include 和lib64 文件

这时候在非root的根目录下的.local(如果没有可以mkdir .local 创建) 软连接或者复制刚刚解压的文件到该目录下，并在~/.bachrc

里面添加如下语句：

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/home/用户名/.local/lib64

然后更新环境就可以使用cudnn7.6