ubuntu16.04系統下CUDA8.0和CUDNN6.0和tensorflow(GPU)的安裝

寒假一過,由於實驗室電腦出了問題,進行了重裝一切。
當然關鍵還在tensorflow的安裝,要知道第一次摸索了一週才裝好的啊。這次特地憑着一個月前的記憶並留心記錄下了tensorflow的安裝步驟,以供包括我你在內的大家學習~(雖然是第二次仍然有以前沒有碰到的問題,難熬啊,安裝成功一定要出去浪一波)

希望對親們有幫助~


第一步 安裝NVIDIA顯卡驅動
1. 打開terminal(怎麼打開?納尼問度娘!),輸入sudo apt-get update 更新系統(NVIDIA比較新的內核版本和ubuntu新版本有一定的對應關係;注意更新之後下次打開系統的時候在ubuntu選擇頁面選擇新的內核的ubuntu,那個是新的?當然是數字大的那個就是啊)
2. 安裝顯卡驅動 打開 系統設置-軟件更新-附加驅動,選擇見圖:

3.這裏重啓電腦!!!
打開NVIDIA X Server Settings,圖像如下就是安裝成功啦(網上其他驗證方法都可以參考啦)

ps:如果中途哪裏弄錯了,還是可以卸載原驅動重新安裝滴,卸載方法:
sudo apt-get remove --purge nvidia --*,重新安裝請看第一步即可。

第二步 安裝CUDA8.0
1.下載CUDA8.0
CUDA 8.0下載地址:
選擇linux的deb版本,選擇如下圖:


2. gcc降低版本
由於這裏gcc5.0以上版本和將要安裝的環境不匹配,而ubuntu16.04自帶gcc5.0以上版本,所以這裏可以先在終端輸入:
gcc --version
從而查看gcc的版本,如果在5.0以下就請調到下一個小步驟。降版本請在終端輸入下面的幾行命令:

再在終端輸入:
gcc --version
結果如下:

3. CUDA8.0的安裝
在下載文件夾裏打開終端。在終端輸入如圖所示的三行命令:


4. 配置環境變量:
先在終端執行如下命令打開配置環境變量的環境:
sudo gedit ~/.bashrc
再在打開的文檔的最低端加上兩行命令:

保存並關閉,執行如下命令使環境生效:
source ~/.bashrc

第三步 安裝CUDNNv6.0
1. 下載CUDNNv6.0,網址鏈接如下:
這裏需要賬戶登陸纔可以下載,所以需要多一步註冊。
CUDNN下載畫面如下:

選擇和我們要配置的匹配的,並打開小的菜單欄,如下圖:

選中並下載上圖中畫下劃線的文件。

2. 安裝和驗證
在下載目錄下打開終端,執行下圖四行代碼進行解壓其中一個壓縮包並配置文件:

接下來解壓剩下的三個安裝包:

下面進行安裝驗證,如圖:

顯示test passed即說明CUDNN安裝成功!!

第四步 安裝tensorflow
1. 首先安裝庫(可以跳過)
sudo apt-get install libcupti-dev

2.pip 安裝(根據自己要安裝的python版本選擇)
$ sudo apt-get install python-pip python-dev # for Python 2.7$ sudo apt-get install python3-pip python3-dev # for Python 3.n

3.tensorflow安裝$ pip install tensorflow-gpu==1.4.0 # Python 2.7; GPU support$ pip3 install tensorflow-gpu # Python 3.n; GPU support
這裏特別注意tensorflow的版本,根據官方文檔(如下圖),在博客裏所說的配置下可以安裝tensorflow1.4.0版本,博主就好幾次因爲安裝了最新的tensorflow導致安裝失敗好幾天找不到原因。

這裏附上官方安裝說明地址:

https://tensorflow.google.cn/install/install_linux


好啦,等待一會就安裝好啦,接下來就是驗證了。

第五步 安裝驗證
在終端輸入 python ,即轉入python輸入,
然後輸入代碼:
[python] view plain copy
  1. import tensorflow as tf  
  2. hello = tf.constant('Hello, Word!')  
  3. sess = tf.Session()  
  4. print(sess.run(hello))  
如果輸出‘Hello, World!’即說明安裝成功~
大功告成啦

參考:

https://tensorflow.google.cn/install/install_linux#ValidateYourInstallation

http://blog.csdn.net/fdqw_sph/article/details/78745375

http://blog.csdn.net/lyc_yongcai/article/details/73244388

http://blog.csdn.net/jiang_z_q/article/details/73264561

https://www.cnblogs.com/villow/p/tensorflow.html

http://blog.csdn.net/zhaoyu106/article/details/52793183

以及一些CUDA和CUDNN的官方手冊

歡迎大家交流學習~
以上

來自一隻剛入坑的小白

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章