美文网首页
ubuntu 16.04 安装CuDnn

ubuntu 16.04 安装CuDnn

作者: fwei | 来源:发表于2017-03-27 22:13 被阅读0次

实验室服务器安装了cuda之后,需要安装cudnn加速,安装记录如下:

  1. 下载对应cuda版本的cudnn,需要注册Nvidia开发者账号。NVIDIA cuDNN下载地址
  2. 解压:tar zxvf cudnn-8.0-linux-x64-v5.1.tgz
    看到如下目录:
cudnn.jpg

3.cuda的安装目录通过which nvcc 查看

nvcc.jpg
cuda安装目录为/usr/loca/cuda-8.0

4 把include/ 和 lib64/目录下的文件拷贝到cuda的安装目录下的include/ 和 lib64里面

$ cd ~/src/cuda #解压后的路径
$ sudo cp -P include/cudnn.h /usr/local/cuda-8.0/include
$ sudo cp -P lib64/libcudnn* /usr/local/cuda-8.0/lib64
$ sudo chmod a+r /usr/local/cuda-8.0/lib64/libcudnn*

参考How can I install CuDNN on Ubuntu 16.04?

相关文章

网友评论

      本文标题:ubuntu 16.04 安装CuDnn

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