2012年12月17日

Theano gpu setting

根據官方網站的設定:

如果要改用gpu而不是使用cpu來計算函數,必須在import theano之前就先設定,方法有兩種:
  1.  在$HOME/.theanorc中設定
  2. 在環境變數THEANO_FLAGS中設定

而在eclipse的開發環境中,如果要對不同的檔案使用不同的設定,使用方法2較有彈性,執行設定方法如下:
  1. 切換到要執行的檔案,選擇上方的Run->Run configrations->Environment->New,然後Name中填入THEANO_FLAGS,values中填入floatX=float32,device=gpu後, 按下方的apply
  2. 因為eclipse似乎無法正確讀入環境變數中的PATH設定,因此要在同一畫面中加入name=PATH, values=usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin:/usr/games:/usr/local/cuda/bin:/usr/local/cuda/bin (依cuda安裝位置而定).

之後在程式當中即可正確使用gpu來計算。
也可用print theano.config來確定設定正確。