机器学习环境配置系列二之cuDNN

Posted 2021-01-14 jaww

1、下载cuDNN

1. 前往: NVIDIA cuDNN home page.
2. 进入下载
3. 勾选Nvidia的协议复选框（流氓的选择，不勾选不能下载）
4. 选择与安装的cuda版本一致的cudnn进行下载.

2、安装cuDNN

• 解压文件包

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

• 拷贝解压缩的文件

$ 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*

　　

3、路径配置（路径配置非常重要，这个决定了cudnn是否可以正确运行）

• 在用户目录下执行命令

vi .bashrc

• 添加内容如下

export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

• 运行命令

source ~/.bashrc