cudaは,CUDA 7.5 Downloads | NVIDIA Developerあたりから落として入れる.
私はネットワークインストーラを選択して入れた.入れ方はダウンロードリンクのすぐ下に書かれた通りにやればOK.
なんか,nvidiaドライバもこれで入る.入ったかどうかは
$ lsmod | grep nv nvidia_uvm 77824 0 nvidia 8544256 1 nvidia_uvm drm 344064 5 i915,drm_kms_helper,nvidia
みたいな感じで入ってればいいんじゃないかしら.
記憶にないけど,なんかnouveauは入ってなかった.
私のマシンはオンボードのIntelグラフィックコントローラとGeForceの両方を搭載しているので,多分最初にubuntu入れる時,グラボを使わなかったんじゃなかろーかと.よく知らんけど.
cuDNNをNVIDIA cuDNN | NVIDIA Developerから落としてきて入れる.
登録は必須っぽいから,仕方なくやる.
cudnnのtar ballを解凍したら,cudaというディレクトリができるので,cudaのインストールパスにコピる.
多分,デフォなら /usr/local/cuda あたりにあるはず.
chainerを入れ直す.
chainerはキャッシュを使うとコンパイルし直してくれないっぽいので,キャッシュを無効化して入れる.
$ pip install chainer --user --no-cache-dir
ローカルに入れない場合は,userオプションを抜いてsu権限でやる.
うまくいけば,pythonから呼び出した時にエラーが出なくなる.
>>> from chainer import cuda >>> cuda.zeros(1) array([ 0.], dtype=float32)