cuDNN 安装

作者: 盲狙小堡垒 | 来源:发表于2017-03-01 14:50 被阅读350次
    前提

    已经安装好了CUDA,并且知道cuda目录(一般是/usr/local/cuda)
    如果不知道可以用以下命令查询

    which nvcc
    

    准备文件

    下载你需要的版本:https://developer.nvidia.com/rdp/cudnn-download

    例如我下载cnDNN5(基于CUDA 8.0 和 Linux)(此操作不一定要在cuda安装目录下,URL地址需要根据你的版本来替换)

    wget https://developer.nvidia.com/rdp/assets/cudnn-7.5-linux-x64-v5.0-ga-tgz
    
    tar -xvf cudnn-7.5-linux-x64-v5.0-ga-tgz
    

    安装(复制文件+创建软连接)

    其实cuDNN的安装就是把几个文件复制到which nvcc显示的目录下。把解压好后的cuda里面的include 和 lib64 都复制到你的CUDA目录下(一般是/usr/local/cuda)
    然后重建一下软连接。

    ln -s libcudnn.so.5.1.3 libcudnn.so.5.1
    ln -s libcudnn.so.5.1 libcudnn.so
    

    OK, now We are done!

    相关文章

      网友评论

        本文标题:cuDNN 安装

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