Ubuntu
CUDA
環境構築
cuDNN
CUDA9.0

CUDA 9.0とcuDNN 7.0をUbuntu 16.04LTSにインストールする

CUDA Toolkit 9.0のインストール

NVIDIAの公式ページからCUDA Toolkit 9.0をダウンロードする。ダウンロードしたフォルダで以下のコマンドを実行する。メタパッケージのcuda-9-0をインストールすると、全てのCUDA ToolkitとDriverパッケージがインストールされる。(CUDA 9.1のインストール方法はこちら

terminal
sudo dpkg -i cuda-repo-ubuntu1604_9.0.*_amd64.deb
sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
sudo apt update
sudo apt install cuda-9-0

パスの設定

以下のechoコマンドでCUDAのパスを.bashrcのファイルの最後に書き込む。その設定をsourceコマンドで読み込む。

terminal
echo -e "\n## CUDA and cuDNN paths"  >> ~/.bashrc
echo 'export PATH=/usr/local/cuda-9.0/bin:${PATH}' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:${LD_LIBRARY_PATH}' >> ~/.bashrc
source ~/.bashrc # CUDAのパスが書き込まれた~/.bashrcを読み込む。
.bashrcに追加される文字列
## CUDA and cuDNN paths
export PATH=/usr/local/cuda-9.0/bin:${PATH}
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:${LD_LIBRARY_PATH}

CUDAのパスが通っているか確認。

terminal
echo $PATH             # 出力に"/usr/local/cuda-9.0/bin"が含まれているか?
echo $LD_LIBRARY_PATH  # 出力に"/usr/local/cuda-9.0/lib64"が含まれているか?
which nvcc             # 出力が"/usr/local/cuda-9.0/bin/nvcc"になっているか?
nvidia-smi             # nvidiaのGPUの情報が表示されているか?

cuDNNのインストール

cuDNN 7.0のインストール

https://developer.nvidia.com/rdp/cudnn-download からcuDNN 7.0 for CUDA 9.1のdebパッケージをダウンロードする(アクセスするにはメンバー登録が必要)。具体的には以下の3個のdebパッケージをダウンロードする。

  • cuDNN v7.0.* Runtime Library for Ubuntu16.04 (Deb)
  • cuDNN v7.0.* Developer Library for Ubuntu16.04 (Deb)
  • cuDNN v7.0.* Code Samples and User Guide for Ubuntu16.04 (Deb)

ダウンロードしたフォルダで以下のコマンドを実行して、cuDNNをインストールする。

terminal
# Install Runtime library
sudo dpkg -i libcudnn7_7.0*+cuda9.0_amd64.deb
# Install developer library
sudo dpkg -i libcudnn7-dev_7.0*+cuda9.0_amd64.deb
# Install code samples and user guide
sudo dpkg -i libcudnn7-doc_7.0*+cuda9.0_amd64.deb 

動作確認

CUDAのサンプルコードを実行して動作確認する。
サンプルコードをホームディレクトリにコピーするシェルスクリプト(cuda-install-samples-9.0.sh)は /usr/local/cuda-9.0/bin/ にあるので、CUDAのパスが通っていればそのファイル名を指定するだけで実行できる。

terminal
cuda-install-samples-9.0.sh ~ # ホームディレクトリにサンプルコードをコピー。
cd ~/NVIDIA_CUDA-9.0_Samples/
make
cd 2_Graphics/volumeRender    # サンプルの実行ファイルがあるディレクトリに移動。
./volumeRender

volume_render.png

アンインストール方法

terminal
sudo apt remove cuda-9-0
sudo apt autoremove
sudo apt remove libcudnn7 libcudnn7-dev libcudnn7-doc
rm -rfv ~/NVIDIA_CUDA-9.0_Samples/ # サンプルコードを消す。
# ~/.bashrcに追加されたPATHの設定をテキストエディタなどで消す。
# ログアウトしてPATHの設定の変更を反映。

参考

https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html#package-manager-metas
https://qiita.com/JeJeNeNo/items/05e148a325192004e2cd