最近看了很多網(wǎng)上的ubuntu安裝tensorflow-gpu版本教程偿曙,感覺很復雜州既。
主要問題其實還是cuda cudnn 和tensorflow-gpu版本之間兼容的問題谜洽,不同的tf對應不同的cudnn和cuda。下面介紹本人親測成功的方案
ubuntu16.04 (系統(tǒng)版本無所謂吴叶,都行)
anaconda 3
gpu? gtx1063
cuda? 9.0??
cudnn 7.0? ? (for cuda 9.0)
安裝顯卡驅動
右上角-->系統(tǒng)設置-->軟件和更新-->附加驅動
然后顯卡驅動勾上阐虚,點應用更改
等下載完后重啟一遍,顯卡驅動就安裝好了蚌卤,下載terminal里面輸入nvidia-smi 來查看
安裝cuda
本次教程用的版本是cuda-9.0
傳送門:cuda下載地址
下載完了之后奥秆,terminal cd到下載目錄下面輸入一下命令進行安裝
sudo dpkg -i cuda-repo-ubuntu1604-9-0-local_9.0.176-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
然后就完成了
這里可以設置環(huán)境變量也可以不設置環(huán)境變量(不用設置也能使用tensorflow-gpu)
還是順便說一下:
export CUDA_HOME=/usr/local/cudaexport PATH=$PATH:$CUDA_HOME/binexport LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
安裝cudnn
這個東西下載之前要注冊登錄一下
傳送門:cudnn下載傳送門
下載如圖:
現(xiàn)在完之后同理,手動解壓出來吧磕洪,不用命令行了吭练,然后cd到解壓目錄下面,輸入一下命令行,將文件復制到cuda的安裝目錄下(這些代碼是復制粘貼的析显,親測有效鲫咽,一行一行執(zhí)行,不要著急):
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
環(huán)境變量配置
在terminal中輸入
sudo gedit ~/.bash_profile?
在文本中輸入一下環(huán)境變量
export LD_LIBRARY_PATH="$LD_LIBRARY_PATH:/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64"
export CUDA_HOME=/usr/local/cuda-9.0
最后保存生效在terminal中輸入 source ~/.bash_profile
安裝tensorflow-gpu
本人安裝了anaconda集成環(huán)境箩绍,直接用pip進行安裝,目前tensorflow-gpu最新版本是1.7尺上,但是安裝好了無法使用材蛛,本人降到1.5版本,可以正常使用
在terminal中輸入?
pip install tensorflow-gpu==1.5
速度太慢的同學自行百度:pip 阿里源 清華源等等加速下載
安裝好了然后利用以下代碼進行測試:
import tensorflowas tf
matrix1 = tf.constant([[3., 3.]])
matrix2 = tf.constant([[2.],[2.]])
product = tf.matmul(matrix1, matrix2)
sess = tf.Session()
result = sess.run(product)
print (result)
效果如圖:
總結
親測可用的版本卑吭,結合了網(wǎng)上很多大神的方法,自己摸索出來的马绝,不需要什么編譯安裝豆赏,不需要什么降版本gcc,省去了很多麻煩富稻,所有安裝包之類的下載之后掷邦,安裝過程不超過20分鐘
本人第一次寫簡書,如果有寫的不好的地方或者更方便的安裝方法椭赋,可以一起討論