#ubuntuの初期設定をする
sudo apt -y update
sudo apt -y upgrade
必要な設定をする。
まだデバッグはしてないが、だいたいあってるはず。
https://github.com/miyamotok0105/setup_tool/blob/master/ubuntu_setup.sh
#nvidiaからdebをDL
ubuntu14.04にcuda8.0を入れるスクリプト。まだ試してない。
https://github.com/miyamotok0105/setup_tool/blob/master/cuda_cudnn_setup.sh
イメージをダウンロード (1.9 GB)
wget https://developer.nvidia.com/compute/cuda/8.0/prod/local_installers/cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64-deb
#cudaのインストール
sudo dpkg -i cuda-repo-ubuntu1404-8-0-local_8.0.44-1_amd64.deb
sudo apt update
sudo apt install cuda
#.bashrcに記述
export CUDA_HOME=/usr/local/cuda-8.0
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:${CUDA_HOME}/lib64
export PATH=$PATH:${CUDA_HOME}/bin
反映
source .bashrc
cudaが動くかチェック
cd /usr/local/cuda-8.0
sudo make
#cuDNNのインストール
Deep Learningに特化したライブラリで畳み込み等の計算を高速に処理してくれるライブラリ。ライブラリを取得するためにはデベロッパー登録が必要。
展開
tar -zxf cudnn-8.0-linux-x64-v5.1.tgz
コピー
sudo cp lib64/* /usr/local/cuda/lib64/
sudo cp include/cudnn.h /usr/local/cuda/include/
#確認
cudaパスが通ってることを確認
which nvcc
nvidia-smi
+-----------------------------------------------------------------------------+
| NVIDIA-SMI xxx.xx Driver Version: xxx.xx |
|-------------------------------+----------------------+----------------------+
| 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 TIT... Off | 0000:00:00.0 Off | N/A |
| 22% 34C P0 70W / 250W | 0MiB / 12206MiB | 0% Default |
+-------------------------------+----------------------+----------------------+
#注意
cudaのビルドがうまくいってない、あるいはパスがうまくいってない場合はDLライブラリのインストールでビルドがこける場合があった。
cuda-8.0でgccの5以上のバージョンをサポートしてないと書いてる。gccバージョンによってビルドエラーもアリエル。
http://blog.mixable.de/nvidia-cuda-on-ubuntu-unsupported-gnu-version-gcc-versions-later-than-5-are-not-supported/
https://plus.google.com/+YoheiKuga/posts/19JtReDP6K5