1. 程式人生 > >[安裝教程] Ubuntu 16.04 LTS + cuda8.0 + cudnn6.0 + python3.5 + tensorfow1.4

[安裝教程] Ubuntu 16.04 LTS + cuda8.0 + cudnn6.0 + python3.5 + tensorfow1.4

寫在前面:請一定按照題目上的版本進行安裝,因為安裝的其他版本之間的搭配均以失敗告終!

配置GPU版本的tensorflow真心時累啊,經歷了版本不對,迴圈登入,等等問題。終於安裝成功,在這記錄一下過程,以備日後再次安裝。

網上有很多安裝教程,但都不是很細緻,我在下面貼了一個連結,寫的很詳細。只有幾個細節需要注意,我寫在下面。

  1. 首先,確認GPU可用,以及是否已安裝驅動。 直接在終端輸入nvidia-smi命令即可,顯示結果如下。請注意Driver Version是否顯示。 在這裡插入圖片描述 如果已經安裝了顯示卡驅動,在cuda的安裝過程中請勿選擇cuda自帶的驅動,要在那一項選擇n。否則可能會導致電腦在開機之後會進入無限登入的迴圈。如果不幸導致無限登入迴圈,請按ctrl+alt+F1
    進入文字模式,解除安裝所有顯示卡驅動,具體可參考這篇部落格。解決無限登入
  2. 如果電腦裝了不止一個版本的python,如2和3。在安裝tensorflow-gpu時請注意pip和pip3的區別,pip為python2安裝tensorflow,而pip3才是為python3安裝。