UbuntuにCUDAをインストールする方法

前回、UbuntuにNVIDIAドライバをインストールしたので、今回はCUDAをインストールしてみました。

前準備

~$ sudo apt update
~$ sudo apt upgrade

CUDAのインストール

~$ sudo dpkg -i cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb
~$ sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
~$ sudo apt-get update
~$ sudo apt-get install cuda
~$ sudo apt-get install nvidia-384
~$ sudo reboot

PATHの設定

~/.bashrcの末尾に以下を追加する。

export PATH=/usr/local/cuda-9.0/bin:${PATH}
export LD_LIBRARY_PATH=usr/local/cuda-9.0/lib64:${LD_LIBRARY_PATH}

PATHの確認

~$ 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の情報が表示される