keras(theano)的GPU模式安裝:很久很久沒寫博客了,上次寫博客應該是一年前的事情,如果會看這篇文章的同志們應該也知道keras是啥,這里不多介紹,今天要寫的keras的,先來一下自己的硬件設備和系統:
ubuntu14.04
GTX-950(我使用)
python2.7
1. 安裝必要的工具vim+git:sudoapt-getinstallvimgit
2. 安裝必要的依賴庫: sudoapt-getinstallpython-numpypython-scipypython-devpython-pippython-noseg++libopenblas-dev
3. 安裝nvida驅動:
首先去下載驅動:http://www.nvidia.cn/Download/index.aspx?lang=cn,下載后得到一個./NVIDIA.run 文件(名字具體看你下載的文件)屏蔽ubuntu原來的驅動:sudo gedit /etc/modprobe.d/blacklist.conf 并依次添加一下內容
blacklist vga16fbblacklist nouveaublacklist rivafbblacklist nvidiafbblacklist rivatv
卸載原來驅動:sudo apt-get –purge remove nvidia-* sudo apt-get –purge remove xserver-xorg-video-nouveau接下來關閉ubuntu圖形界面安裝nvida驅動: sudo service lightdm stop然后切換到tty1:Ctrl+Alt+F1登錄root用戶接下來安裝下載下來的驅動: ./NVIDIA.run開始安裝重新啟動X-Window:sudo service lightdm start lspci|grep-invidia 檢查是否安裝成功
4. 安裝cuda:
選擇cuda:http://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/ (我用的6.5,因為我發現安裝7.5后和我顯卡驅動無法兼容,具體大家可以使用新的版本) 5.安裝Anaconda:這個玩意包含了蠻多python有用的lib(numpy,scipy,pyyaml,hdf5),所以可以減少你很多lib的安裝。下載地址:http://www.continuum.io/downloads#all。一條命令安裝好: 7. 安裝keras:sudo pip install keras 。 shell中鍵入python,之后輸入from keras.models import Sequential PS:我知道基本上csdn的安裝流程都不能安裝成功,所以有問題就留言碰吧。 缺少yaml的解決方法 最后keras有一個style-transfer的demo給大家看一下有點努力的動力吧。下面是keras的4次迭代的結果。
下載cuda: wgethttp://developer.download.nvidia.com/compute/cuda/repos/ubuntu1404/x86_64/cuda-repo-ubuntu1404_6.5-14_amd64.debsudodpkg-i cuda-repo-
6. 安裝theano,這一步需要說明一下,我習慣用conda來虛擬一個環境后再進行操作,可以參考這個:http://www.tuicool.com/articles/vyyA7rB sudo apt-get install python-numpy python-scipy python-dev python-pip python-nose g++ libopenblas-dev git sudo pip install Theanocdwget http://pyyaml.org/download/libyaml/yaml-0.1.4.tar.gztar xzvf yaml-0.1.4.tar.gzcd yaml-0.1.4./configure --prefix=/usr/localmakesudo make install
歡迎轉載:http://www.kanwencang.com/bangong/20161228/78729.html
文章列表