こんにちは～
機械学習の勉強のため、Tensor flowを触りたいということで、二台目のノートPCにubuntuを入れました！
なぜubuntuかといえば Tensor flowのGPU機能がlinuxしか対応していないからです。。。(2016年4月時点)
Chainerを触ったことがありますが、GPUあるなしで性能に雲泥の差がでるのは間違いないので、本気でGPU対応するように挑戦してきました。
現状、Tensor flowはCUDA 7.0が必要です
色々調べてみたら、CUDA 7.0をインストールする記事は結構ありましたが、NVidia Optimusを搭載したPCでCUDAを入れる方法はありませんでした。
なので、色々記事を参考にして、どうにかCUDAをいれました。。
ここまでの道のりがものすごく大変でしたので、今後のために備忘録として紹介します！
環境は次のとおりです
- PC: ASUS K55VD
- OS: Ubuntu 14.04
- GPU: Geforce 610M(NVidia Optimus対応)
bumblebeeインストール
NVidia OptimusでGPUを使用する場合は、ただ単にドライバのインストールしても動きません！
そこで、NVidia Optimus環境でもGPUを使うためのツールとして, bumblebeeがあります！
これをインストールしましょう！
# bumblebeeのリポジトリ追加＋最新版へ更新
sudo add-apt-repository ppa:bumblebee/stable
sudo apt-get update
# bumblebee一式のインストール(公式ページでのubuntu14.04のインストール方法を参考)
sudo apt-get install bumblebee bumblebee-nvidia primus linux-headers-generic
nvidia-352 nvidia-settingsインストール
CUDA7.0を入れるため、nvidiaのドライバは352以上が必要です。ここでインストールしましょう！
sudo apt-get install nvidia-352 nvidia-settings
/etc/bumblebee/bumblebee.confでnvidia-352を設定
nvidia-352をbumblebeeで利用するようにしましょう！
sudo vim /etc/bumblebee/bumblebee.conf
# 次のように変更します
22行: Driver=nvidia
55行: KernelDriver=nvidia-352
58行: LibraryPath=/usr/lib/nvidia-352:/usr/lib32/nvidia-352
61行: XorgModulePath=/usr/lib/nvidia-352/xorg,/usr/lib/xorg/modules
/etc/modprobe.d/bumblebee.confでnvidia-352をブラックリストに入れることでロードを抑制
nvidia-352をインストールした場合、bumblebeeを起動する前にnvidia-352を有効になってしまいます。そうなると、bumblebeeが起動せず、Ubuntuが正しく起動しません（原因がわかるまで、UbuntuのGUI起動後真っ暗になっていて大変でした。。。）
そこで、/etc/modprobe.d/bumblebee.confでnvidia-352をブラックリストに入れることでロードを抑制させます！
sudo vim /etc/modprobe.d/bumblebee.conf
# 以下を追加
blacklist nvidia-352
blacklist nvidia-352-updates
blacklist nvidia-experimental-352
これで、Nvidiaの新しいドライバをbumblebee上で動かせるようになりました。
CUDAのダウンロード
つぎはCUDAを入れていきます。
まずは、CUDAをダウンロードします。debパッケージでもインストールできますが、細かいオプションが利かないため、runパッケージをダウンロードします！
下記ページからubuntu14.04のrunパッケージを選択してダウンロードするか、
あるいは直接落とします。
wget http://developer.download.nvidia.com/compute/cuda/7_0/Prod/local_installers/cuda_7.0.28_linux.run
落としたら、ホームディレクトリ（~/）へ移動させます
CUDAのインストール
何も考えずにインストールすると、OpenGLライブラリもインストールされます。このライブラリがインストールすることによって、Ubuntuがログインできなくなります！（アンインストールしても元に戻らなかったため、ここでUbuntuを再インストールするハメになりました。。。）
なので、OpenGLライブラリをいれずにインストールします！
./cuda_7.0.28_linux.run --no-opengl-libs
以下指示が出ますが、つぎのように選択します
# 当然accept
Do you accept the previously read EULA? (accept/decline/quit): accept
# 必要らしいので, y
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 346.46? ((y)es/(n)o/(q)uit): y
# CUDA本体なので, y
Install the CUDA 7.0 Toolkit? ((y)es/(n)o/(q)uit): y
# CUDAの入れる場所を尋ねられます。ここはデフォルトのままなので、そのままENTER
Enter Toolkit Location [ default is /usr/local/cuda-7.0 ]:
# sudoで実行しない場合これがでます。sudoで実行させないとインストールできないので、 y
/usr/local/cuda-7.0 is not writable.
Do you wish to run the installation with 'sudo'? ((y)es/(n)o): y
Please enter your password:
# シンボリックリンクをはるか尋ねられますが、実効に必要なので y
Do you want to install a symbolic link at /usr/local/cuda? ((y)es/(n)o/(q)uit): y
# サンプルを入れるかどうか尋ねられますが、動作確認したいので y
Install the CUDA 7.0 Samples? ((y)es/(n)o/(q)uit): y
# サンプルの入れる場所はデフォルトにしときます
Enter CUDA Samples Location [ default is /home/faithnh ]:
問題なければこれでインストールできます！
最後にこの指示がでますので、
Please make sure that
- PATH includes /usr/local/cuda-7.0/bin
- LD_LIBRARY_PATH includes /usr/local/cuda-7.0/lib64, or, add /usr/local/cuda-7.0/lib64 to /etc/ld.so.conf and run ldconfig as root
~/.bashrc上に設定しときましょう！
vim ~/.bashrc
# 以下を追加
# for CUDA
export PATH=$PATH:/usr/local/cuda-7.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-7.0/lib64
そして、設定を反映させます。
source ~/.bashrc
CUDAサンプルをビルド、実行
上記手順でインストールしたら、ホームディレクトリに配置されますので、サンプルにアクセスし、makeでビルドします。
makeするには、gccなどの環境が必要なので、開発環境一式をインストールしておきます
sudo apt-get install build-essential
そのあと、makeしてビルドしていきます。ビルドは大変時間がかかります（ここで1時間ほどとられました）
cd ~/NVIDIA_CUDA-7.0_Samples
make
最後にFinished building CUDA samplesがでればOKです！
実行してみるものの、問題が・・・
マージソートのサンプルを実行してみます。
cd ~/NVIDIA_CUDA-7.0_Samples/bin/x86_64/linux/release
./mergeSort
./mergeSort Starting...
CUDA error at ../../common/inc/helper_cuda.h:1105 code=38(cudaErrorNoDevice) "cudaGetDeviceCount(&device_count)"
あらら、問題が。。。
色々調べたところ、bumblebeeでは、デフォルトGPUは有効になっていないため、GPUは認識しないようです。
GPUを有効にして実行させるためにはoptirunコマンドを挟んで実行しなければなりません。
使い方はこんな感じです。
optirun [オプション] アプリ実行コマンド
オプションはなしでも実行できるので実質こうなります。
optirun アプリ実行コマンド
それでは実行してみましょう。
optirun ./mergeSort
./mergeSort Starting...modprobe: FATAL: Module nvidia-uvm not found.
CUDA error at ../../common/inc/helper_cuda.h:1105 code=30(cudaErrorUnknown) "cudaGetDeviceCount(&device_count)"
今度は別の問題がでました、、、nvidia-uvmがないと怒られてます。
つぎは、nvidia-uvmを有効にして対処していきましょう。
nvidia-uvmを有効化
CUDAを動かすにはnvidia-uvmが必要だが、bumblebeeで動作する場合、なぜか有効になっていないため、対処していきます。
まずはbumblebee.confをこのように修正していきます。nvidia-352-uvmをnvidia-uvmとして利用できるようにします。
sudo vim /etc/modprobe.d/bumblebee.conf
alias nvidia-uvm nvidia-352-uvm
設定を反映させ、nvidia-uvmを有効にしていきます！
sudo updatedb
sudo optirun modprobe nvidia_uvm
再度実行
再度マージソートのサンプルを実行してみます。
optirun ./mergeSort
./mergeSort Starting...modprobe: FATAL: Module nvidia-uvm not found.
CUDA error at ../../common/inc/helper_cuda.h:1105 code=30(cudaErrorUnknown) "cudaGetDeviceCount(&device_count)"
うーん、まだ動かないな。。。
あ、よく考えたらnvidia_uvmではsudoで実行してたので、sudo入れたら動くのでは。。。。
と思い、sudo を入れてみました。
cd ~/NVIDIA_CUDA-7.0_Samples/bin/x86_64/linux/release
sudo optirun ./mergeSort
./mergeSort Starting...
GPU Device 0: "GeForce 610M" with compute capability 2.1
Allocating and initializing host arrays...
Allocating and initializing CUDA arrays...
Initializing GPU merge sort...
Running GPU merge sort...
Time: 1037.418945 ms
Reading back GPU merge sort results...
Inspecting the results...
...inspecting keys array: OK
...inspecting keys and values array: OK
...stability property: stable!
Shutting down...
う、うごいた～！
なぜsudo入れないといけないのかがよく分かりませんが、これで動きました！（後で理由を調べたいと思います）
終わりに
CUDAを入れるだけでものすごい労力がかかってしまったため、残念ながらcuDDN以降は紹介できませんでした
が、Tensor flowが無事に入れた暁に紹介していきたいと思います！
参考記事
http://ry0.github.io/blog/2015/08/12/ubuntu-nvidia-cuda-7.0/
http://ubuntuforums.org/showthread.php?t=2262882&page=7
http://www.linuxveda.com/2015/07/16/how-to-install-drivers-for-nvidia-optimus-cards/
https://ask.fedoraproject.org/en/question/77633/install-nvidia-driver-for-cuda-without-turn-it-on-for-xwindows/