CUDA & Ubuntu16.04 環境構築

cudaの環境構築でかなり右往左往したので,メモ.

nvidiaドライバ

add-apt-repositoryでnvidiaのドライバをインストールする方法をよく見かけるが,私のやり方が悪いのか失敗した.
(多分,GFGTX-960なのにnvidia-375をインストールしてたから.
 でも,公式サイトでは,nvidia-375がレコメンドされたんだけどなぁ。。。)

まず,環境設定のソフトウェアとアップデートからインストールした方が確実であった.

[環境設定] -> [ソフトウェアとアップデート] -> [追加のドライバ] -> もともとubuntuにインストールされているドライバからnvidia-375に変更.

cuda

公式サイトからcuda-toolkitをインストール.

https://developer.nvidia.com/cuda-toolkit

debパッケージがおすすめ.ubuntuに初めからインストールされているnouveauのドライバーをblacklistに追加しないといけなくなるから.

その後,

sudo dpkg -i ダウンロードファイル.deb
sudo apt update
sudo apt install cuda

実行確認

nvidia-smi

次のようなものが出力されたらOK




jun-i@jun-i:~$ nvidia-smi
Tue Dec 27 15:05:11 2016

                                                                                                                                                            • +
NVIDIA-SMI 367.57 Driver Version: 367.57
-------------------------------+----------------------+----------------------+
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 960 Off 0000:01:00.0 On N/A
30% 23C P8 10W / 120W 251MiB / 1993MiB 1% Default
                                                                                                                                                            • +

                                                                                                                                                            • +
Processes: GPU Memory
GPU PID Type Process name Usage
=============================================================================
0 976 G /usr/lib/xorg/Xorg 181MiB
0 1496 G compiz 68MiB
                                                                                                                                                            • +