安装cuDNN到Ubuntu系统

2 年 ago

新, 韵

1 minute

メモとして残しておきます。

cuDNNとは

NVIDIA发布了用于深度学习的库。使用这个库可以提高Caffe和Chainer等深度学习软件的速度。

cuDNN的安装

请从以下链接下载。

要访问此内容，需要进行会员注册。

cuDNN可以在NVIDIA的开发者网站上下载。

具体而言，需要下载三个Deb软件包：
– 运行库
– 开发库
– 代码示例和用户指南

CUDA可在NVIDIA官方网站上下载，网址为https://developer.nvidia.com/cuda-80-ga2-download-archive

~$ sudo dpkg -i libcudnn7_7.4.1.5-1+cuda9.0_amd64.deb
~$ sudo dpkg -i libcudnn7-dev_7.4.1.5-1+cuda9.0_amd64.deb
~$ sudo dpkg -i libcudnn7-doc_7.4.1.5-1+cuda9.0_amd64.deb

将cuDNN库复制到已安装CUDA的目录中。

~$ tar xvf cudnn-9.0-linux-x64-v7.4.15.tgr
~$ sudo cp -a cuda/include/cudnn.h /usr/local/cuda/include/
~$ sudo cp -a cuda/lib64/libcudnn* /usr/local/cuda/lib64/
~$ sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
~$ cp -r /usr/src/cudnn_samples_v7/ $HOME
~$ cd $HOME/cudnn_samples_v7/mnistCUDNN
~$ make clean && make

由于出现错误，因此请输入以下命令。

~$ sudo apt install gcc-6
~$ sudo apt install g++-6

~$ sudo ln -s /usr/bin/gcc-6 /usr/local/cuda/bin/gcc
~$ sudo ln -s /usr/bin/g++-6 /usr/local/cuda/bin/g++
~$ cd $HOME/cudnn_samples_v7/mnistCUDNN
~$ make clean && make
./mnistCUDNN

+αとしてcudaのversionアップの方法である

~$ sudo apt-key add /var/cuda-repo-9-2-local/7fa2af80.pub
~$ sudo apt-get update
~$ sudo apt-get install cuda

*/usr/local/にインストールしたcudaがあります。

~/.bashrc の内容を現在のversionに書き換え

~$ sudo vim ~/.bashrc
~$ source ~/.bashrc

#技术文章