最近要训练博客的标签分类系统,因为使用tensorflow,看了一下训练服务器已经安装好cuda,但是没有安装cudnn,于是从官网将cudnn的tar文件下载下来,参考官网安装教程一顿操作:

​​​​​​​$ tar -xzvf cudnn-x.x-linux-x64-v8.x.x.x.tgz
$ sudo cp cuda/include/cudnn*.h /usr/local/cuda/include 
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64 
$ sudo chmod a+r /usr/local/cuda/include/cudnn*.h 
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

安装完成之后,运行训练程序,报错Could not load dynamic library 'libcudnn.so.8',说明安装失败。

突然想起来之前安装时都是复制到cuda的安装目录/usr/local/cuda-11.1,而不是/usr/local/cuda,这里cuda目录是cuda-11.1的软连接。

$ sudo cp cuda/include/cudnn*.h /usr/local/cuda-11.1/include
$ sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-11.1/lib64
$ sudo chmod a+r /usr/local/cuda-11.1/include/cudnn*.h 
$ sudo chmod a+r /usr/local/cuda-11.1/lib64/libcudnn*

执行完之后完美运行。

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐