theano with GPU

theano with GPU

タグ theano


Theanoは様々な依存パッケージが存在していて一発で入れることが非常に面倒なつくりになっている(っぽい)

THEANO_FLAGS="device=cuda,mode=FAST_RUN,floatX=float32" python train.py
  1. githubからtheanoの最新版をインストールする
  2. githubからpygpuの最新版をインストールする
  3. NVidiaから最新のcuda 8.0 overをインストールする
  4. cudannをインストールし、pathにnvccを追加、/usr/local/libに libcuda系をインストールして、ldconfigして/usr/local/includeに cudaのヘッダーをインストールする