首先在{HOME}下建立.theanorc檔案,內容如下:
[global]
floatX = float32
device = gpu0
[nvcc]
fastmath = True
存檔後,退出至{HOME}中。
輸入nvcc指令測試cuda套件是否已經安裝完成。
然後到eclipse隨便建立一個python檔,import theano後,執行該程式,
此時可能會出現以下錯誤:
ERROR (theano.sandbox.cuda): nvcc compiler not found on $PATH. Check your nvcc installation and try again.
這是因為eclipse的pydev套件使用的PATH變數與linux中不同,
用以下指令修正:
window->preferences->PyDev->Interpreter-Python->Environment->New-> Name: PATH, Value: ${env_var:PATH}:/usr/local/cuda-5.0/bin
設定完成後,可在python檔使用
import os
print os.environ['PATH']
確定環境變數修改完成。