cuDNN 5.1をapt-getでUbuntu 16.04 LTSにインストールしてみた


Ubuntu 16.04 LTSに対応したNVIDIA Machine Learningリポジトリを追加して、apt-getでcuDNN 5.1をインストールしてみました。

NVIDIA Machine Learningリポジトリは、DIGITS、NVcaffe、cuDNN等を含むリポジトリで、これまでUbuntu 14.04 LTS版が用意されていましたが、最近Ubuntu 16.04 LTS版が追加されました。

現在のUbuntu 16.04 LTS版のリポジトリに含まれているパッケージは、DIGITS 5.0、CUDA 8.0用のNVcaffe 0.15.13やcuDNN 5.1.10です(また、NVcaffe等にリンクされていないと思いますがcuDNN 6.0も)。

なお、CUDA 8.0のインストールについては、以下の記事をご覧ください。

CUDA 8.0とNVIDIA DriverをUbuntu 16.04 LTSにインストールしてみた
CUDA 8.0とNVIDIA DriverをUbuntu 16.04 LTSにインストールしてみました。
スポンサーリンク

cuDNN 5.1

準備

ここ(https://github.com/NVIDIA/DIGITS/blob/digits-5.0/docs/UbuntuInstall.md)に記載されているUbuntu 16.04版のDIGITS 5.0と同じ方法で準備を行います。

$ ML_REPO_PKG=http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/nvidia-machine-learning-repo-ubuntu1604_1.0.0-1_amd64.deb
$ wget "$ML_REPO_PKG" -O /tmp/ml-repo.deb
$ sudo dpkg -i /tmp/ml-repo.deb
$ rm -f /tmp/ml-repo.deb
$ sudo apt-get update

なお、sudo dpkg -i /tmp/ml-repo.debの後で、以下のメッセージが出ましたが、ここ(http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/)に、7fa2af80.pubにアップロードされていないことが原因のようです。

gpg: 有効なOpenPGPデータが見つかりません。
Failed to add GPGKEY at http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64/7fa2af80.pub to apt keys.

インストール

DIGITS 5.0をインストールする場合は、以下のコマンドを実行します。

$ sudo apt-get install digits

今回は、cuDNN 5.1(libcudnn5-devと追加パッケージのlibcudnn5)だけをインストールするので、以下のコマンドを実行します。

$ sudo apt-get install libcudnn5-dev

なお、インストール時点のcuDNNのバージョンは、5.1.10でした。
また、cuDNN 5.1のruntimeだけで良ければ、以下のコマンドを実行します。

$ sudo apt-get install libcudnn5

なお、cudnn.hlibcudnn*がインストールされる場所が、/usr/local/cudaではなく、以下のようになっていますので、ChainerでcuDNNを利用する場合は、ここ(http://docs.chainer.org/en/latest/install.html#install-chainer-with-cuda-and-cudnn)を参照して、環境変数を追加する必要があります。

$ find /usr/ -name *cudnn*
/usr/include/cudnn.h
/usr/include/x86_64-linux-gnu/cudnn_v5.h
/usr/lib/x86_64-linux-gnu/libcudnn.so.5.1.10
/usr/lib/x86_64-linux-gnu/libcudnn.so
/usr/lib/x86_64-linux-gnu/libcudnn_static.a
/usr/lib/x86_64-linux-gnu/libcudnn_static_v5.a
/usr/lib/x86_64-linux-gnu/libcudnn.so.5
...

まとめ

Ubuntu 16.04に対応したNVIDIA Machine Learningリポジトリの中にcuDNN 5.1(とcuDNN 6.0)が含まれているので、リポジトリを追加して、apt-getでcuDNN 5.1をインストールしてみました。