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

1、下载cuDNN

前往: NVIDIA cuDNN home page.
进入下载
勾选Nvidia的协议复选框（流氓的选择，不勾选不能下载）
选择与安装的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

巴特西

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

最新文章

热门文章