Ubuntu Desktop 18.04にcuDNN 7.6.5をインストールする!

CUDA 10.2用のデープラーニング用のライブラリcuDNN 7.6.5をインストールします。

使用した環境

  • Intel® Core™ i5-7500
  • GeForce GTX 1050 Ti
  • Ubuntu 18.04 LTS Desktop (18.04.3)
  • CUDA 10.2
  • cuDNN 7.6.5

CUDAのインストール

CUDAをインストールしていない場合は、以下の記事を参考にインストールします。

cuDNN 7.6.5のダウンロード

cuDNN 7.6.5のインストール用モジュールを、Nvidiaのホームページ「NVIDIA cuDNN」から「Download cuDNN」ボタンをクリックして画面遷移し、ユーザ登録を行い、ダウンロードページにログインします。インストール済みのCUDAのバージョン用のページに遷移して、以下を選択して、ダウンロードします。

  • cuDNN Runtime Library for Ubuntu18.04 (Deb)
  • cuDNN Developer Library for Ubuntu18.04 (Deb)
  • cuDNN Code Samples and User Guide for Ubuntu18.04 (Deb)

cuDNN 7.6.5のインストールの実行

以下はCUDA 10.2での例ですが、dpkg -iのコマンドを実行して、cuDNN 7.6.5をインストールします。

$ cd $HOME/Downloads
$ ls -1 libcudnn*
libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn7-dev_7.6.5.32-1+cuda10.2_amd64.deb
$ sudo dpkg -i libcudnn7-doc_7.6.5.32-1+cuda10.2_amd64.deb

cuDNN 7.6.5のインストールの検証

cuDNN 7.6.5のインストールの検証を行います。最後に、「Test passed!」と表示されることを確認します。

$ cp -r /usr/src/cudnn_samples_v7/ $HOME
$ cd  $HOME/cudnn_samples_v7/mnistCUDNN
$ make clean && make
$ ./mnistCUDNN

おわりに

CUDAのデープラーニング用のライブラリcuDNN 7.6.5をインストールしました。特につまずくようなところはないと思います。