cudnn的安装和更换版本

作者: 随波逐流的亚瑟王 | 来源:发表于2018-07-10 15:46 被阅读27次

cudnn的下载 

https://developer.nvidia.com/rdp/cudnn-archive

library就是 tar。。。

然后 其实很简单 按照官网操作就好:

1. Navigate to your directory containing the cuDNN Tar file.

2. Unzip the cuDNN package.

$ tar -xzvf cudnn-9.0-linux-x64-v7.tgz

3. Copy the following files into the CUDA Toolkit directory.

$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include

$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64

$ sudo chmod a+r /usr/local/cuda/include/cudnn.h

/usr/local/cuda/lib64/libcudnn*

其实恶心的是 要换版本

解压出来的lib64下面有3个so文件。分别是 libcudnn.so 和 libcudnn.so.5以及 libcudnn.so.5.1.3文件。 并且这3个点so文件大小都一样。其实都是软连接!libcudnn.so链接到libcudnn.so.5,而libcudnn.so.5.又链接到libcudnn.so.5.1.3。 正真的文件只有libcudnn.so.5.1.3,因此我们要将/usr/local/lib64下的以前的这样的链接替换掉。由于装cuda时,比如我装的是cuda8.0,那么在/usr/local/下会生成cuda-8.0文件夹,以及一个cuda文件夹,cuda是软链接到cuda-8.0的,所以这两个文件夹可以看成一个。往任意一个文件夹中添加东西,另一个文件夹都会有相同的东西。

cd/usr/local/cuda/lib64

ll  就是LL

出现:

-rw-r--r--1root root7201928月162015libcudart_static.a

lrwxrwxrwx1root root1510月322:39libcudnn.so -> libcudnn.so.5.0*

lrwxrwxrwx1root root1710月418:34libcudnn.so.5-> libcudnn.so.5.0.

我们需要把以前的这3个文件删除。libcudnn.so和libcudnn.so.5以及libcudnn.so.5.0

sudo rm libcudnn.so.5libcudnn.so.5.0*

sudo rm libcudnn.so

然后自己创建类似的链接。由于5.1已经被链接到了5.1.3上了。所以我们要删掉以前的.so文件,然后创建新的.so文件,并且链接到5.1上。

sudo ln -s libcudnn.so.5.1 libcudnn.so

ll

最后出现

lrwxrwxrwx1root root1510月622:40libcudnn.so -> libcudnn.so.5.1*

lrwxrwxrwx1root root1710月622:40libcudnn.so.5.1-> libcudnn.so.5.1.3*

-rwxr-xr-x1root root6069670410月622:37libcudnn.so.5.1.3*

-rwxrwxrwx1root root5971599010月622:37libcudnn_static.a*

相关文章

网友评论

    本文标题:cudnn的安装和更换版本

    本文链接:https://www.haomeiwen.com/subject/anzfpftx.html