Ubuntu 16.04でCUDA 9.0からCUDA 8.0にバージョンダウンする

Ubuntu 16.04でCUDA 9.0からCUDA 8.0にバージョンダウンしてみたので、その備忘録。

1. CUDAのインストールガイド

「CUDA」(CUDA Toolkit)と「cuDNN」のインストール方法は、検索すると大量にヒットしますが、「CUDA」のインストールには、OSやバージョンなどの条件に応じて様々な作業手順が必要になるため、以下の公式のクイックスタートガイドを読むのがベストです。

・CUDAクイックスタートガイド

2. CUDAの削除

CUDAのバージョンダウンする際には、インストール前にCUDA一式をアンインストールする必要があります。
アンインストールするコマンドは次の通りです。

$ sudo apt purge cuda*
$ sudo apt purge nvidia-cuda-*
$ sudo apt purge libcuda*
$ sudo apt autoremove
$ reboot

3. CUDAのインストール

(1)CUDAのインストール用ページを開く
最新のCUDAのインストール用ページは以下になります。

・CUDA Toolkit 10.1 Update 1 Download | NVIDIA Developer

同ページの「Legacy Release」で、昔のバージョンのCUDAを選択できます。今回は「CUDA Toolkit 8.0 GA1」を選択しました。

(2)debファイルのダウンロード
インストール用ページでは、OS、アーキテクチャ、ディストリビューション、バージョンをインストール先のシステムにあわせて選択します。
そして、インストール種別は、「deb(local)」または「deb(network)」を選びます。「runfile」はインストール手順が複雑なので、「deb」が良さそうです。
Downloadボタンを押して、debファイルをダウンロードします。

(3)CUDAのインストール
ダウンロードしたdebファイルのあるフォルダで、
インストール用ページに書いてあるコマンドを入力します。
「Linux/x86_64/Ubuntu/16.04/deb(network)」を選択した時は、次のようになります。

$ sudo dpkg -i cuda-repo-ubuntu1604_8.0.44-1_amd64.deb
$ sudo apt-get update
$ sudo apt-get install cuda-8-0

(4)OSのリブート

(5)「~/.bashrc」の「PATH」「LD_LIBRARY_PATH」「LIBRARY_PATH」のパスを変更
/usr/local/に新しいCUDAがインストールされているので、それをパスに反映させます。

export PATH="/usr/local/cuda-8.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH"
export LIBRARY_PATH="/usr/local/cuda-8.0/include:$LIBRARY_PATH"

編集後、以下のコマンドで更新します。

$ source ~/.bashrc

4. cuDNNのインストール

(1)cuDNNの圧縮ファイルのダウンロード
以下のサイトから、cuDNNの圧縮ファイルをダウンロードします。
同ページの「Archived cuDNN Release」で昔のバージョンのcuDNNの圧縮ファイルをダウンロードできます。今回は、「cuDNN v5.1 for CUDA 8.0」をダウンロードしました。

・cuDNN Download | NVIDIA Developer
https://developer.nvidia.com/rdp/cudnn-download

(2)cuDNNの圧縮ファイルを解凍し、CUDAのインストール先フォルダにコピー
cuDNNの圧縮ファイルを解凍し、ライブラリをCUDAのインストール先フォルダにコピーします。

$ tar xzvf cudnn-8.0-linux-x64-v5.1.tgz
$ sudo cp -a cuda/lib64/* /usr/local/cuda-8.0/lib64/
$ sudo cp -a cuda/include/* /usr/local/cuda-8.0/include/
$ sudo ldconfig

5. インストールの確認

以下のコマンドでインストールされたことを確認できます。

◎CUDAのdebパッケージの検索

$ dpkg -l | grep cuda

◎CUDAのバージョン

$ nvcc -V

◎cuDNNのバージョン確認

$ cat /usr/local/cuda-8.0/include/cudnn.h | grep CUDNN_MAJOR -A 2

※「/usr/local/cuda-8.0」はインストール先パス