【Ubuntu Server 16.04 LTS】機械学習用にPCを組んでみた その2（GTX-1080TiドライバとCUDA9.0をインストール）

GTX-1080Tiドライバをインストール

$ sudo add-apt-repository ppa:graphics-drivers/ppa
$ sudo apt update
$ sudo apt install nvidia-381
$ nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 384.90                 Driver Version: 384.90                    |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|===============================+======================+======================|
|   0  GeForce GTX 108...  Off  | 00000000:02:00.0 Off |                  N/A |
|  0%   42C    P5    26W / 275W |      0MiB / 11169MiB |      2%      Default |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                       GPU Memory |
|  GPU       PID   Type   Process name                             Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

上手くいけばGPUの情報が表示されるはずです。

CUDA9.0をインストール ¹

ダウンロードページ ※wgetに指定するURLを調べます
プラットフォームに合わせて選択
例）Linux >> x86_64 >> Ubuntu >> 16.04 >> deb(network)
あとは表示された手順に従います。

$ wget developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
$ sudo dpkg -i cuda-repo-ubuntu1604_9.0.176-1_amd64.deb
$ sudo apt-key adv --fetch-keys http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1604/x86_64/7fa2af80.pub
$ sudo apt-get update
$ sudo apt-get install cuda

cuDNN v7.0をインストール

ダウンロードページ ※登録が必要です

$ tar -xzvf cudnn-9.0-linux-x64-v7.tgz
$ sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
$ sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
$ sudo rm /usr/local/cuda/include/cudnn.h
$ sudo rm /usr/local/cuda/lib64/libcudnn*

---

1. 後々気づきましたがこの後導入するtensorflowにはCUDA9.0が対応していないためエラーになるという情報（リンク先）を見ました。が、現状問題ないです。。。（？） ↩