theano GPU

http://deeplearning.net/software/theano/tutorial/using_gpu.html

1.安裝cuda
下載:https://developer.nvidia.com/cuda-downloads
2.添加環境變量

$ export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
$ export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

3.切換用戶主目錄下

cd ~
vim .theanorc
添加配置:
[global]
device = gpu
floatX = float32
[cuda]
root = /usr/local/cuda

4.運行測試代碼
http://deeplearning.net/software/theano/tutorial/using_gpu.html

發佈了53 篇原創文章 · 獲贊 8 · 訪問量 10萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章