ubuntu安裝TensorFlow-GPU教程

本文的安裝環(huán)境是:ubuntu16.04、CUDA8.0、Cudnn6.0,tensorflow1.3
在這里提供下面所需要的安裝文件供大家使用
鏈接:http://pan.baidu.com/s/1c2atLMo 密碼:78qm

下面的命定不要直接復(fù)制粘貼,要根據(jù)個(gè)人文件路徑的實(shí)際情況。

安裝NVIDIA驅(qū)動(dòng)

先去官網(wǎng)查看適合自己GPU版本的驅(qū)動(dòng):http://www.nvidia.com/Download/index.aspx?lang=en-us
以GTX1080Ti為例:查到其對(duì)應(yīng)的版本號(hào)為:384



得到版本號(hào)之后就是執(zhí)行下面的命定來(lái)更新顯卡驅(qū)動(dòng):

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-(自己的版本號(hào))
sudo apt-get install mesa-common-dev
sudo apt-get install freeglut3-dev

執(zhí)行成功后,重啟電腦,打開(kāi)終端,輸入
nvidia-smi
如果沒(méi)有報(bào)錯(cuò)并且能夠顯示,則說(shuō)明你的驅(qū)動(dòng)更新成功了。

安裝CUDA

先去下載CUDA8.0的安裝文件https://developer.nvidia.com/cuda-80-ga2-download-archive
然后再執(zhí)行以下命定

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
sudo apt-get update
sudo apt-get install cuda
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-cublas-performance-update_8.0.61-1_amd64.deb
sudo apt-get update 

到這里就裝完了CUDA

安裝Cudnn

下載地址:https://developer.nvidia.com/cudnn
再執(zhí)行命定

tar -zxf cudnn-8.0-linux-x64-v6.0.tgz
cd cuda
sudo cp include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp lib64/libcudnn* /usr/local/cuda-8.0/lib64
tar -zxf cudnn-8.0-linux-x64-v6.0.tgz
cd cuda
sudo cp include/cudnn.h /usr/local/cuda-8.0/include/
sudo cp lib64/libcudnn* /usr/local/cuda-8.0/lib64 

安裝Anconada

下載Anconada文件
然后執(zhí)行命定
bash (Anconada的安裝文件名)

安裝TensorFlow

找到對(duì)應(yīng)版本,然后執(zhí)行命定
(在這之前可能要測(cè)試下自己的電腦上安裝了pip沒(méi),如果沒(méi)有的話,請(qǐng)安裝)

pip install (TensorFlow的安裝文件)

然后就可以在自己機(jī)子上測(cè)試是否安裝成功了
先在終端進(jìn)入python
然后輸入

import tensorflow as tf

如果沒(méi)報(bào)錯(cuò),則說(shuō)明安裝成功: congratulations!

參考文獻(xiàn)

感謝前輩的帖子
http://www.linuxidc.com/Linux/2017-01/139321.htm
http://118.89.51.133/?p=41

番外

本人在安裝的過(guò)程中發(fā)現(xiàn),在上面的步驟進(jìn)行到安裝Anconada后
直接輸入命定:

conda install tensorflow-gpu

也能成功安裝
另外安裝keras也能用類(lèi)似命定

conda install keras-gpu

只能說(shuō)Anconada真的很強(qiáng)大

最后編輯于
?著作權(quán)歸作者所有,轉(zhuǎn)載或內(nèi)容合作請(qǐng)聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時(shí)請(qǐng)結(jié)合常識(shí)與多方信息審慎甄別。
平臺(tái)聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點(diǎn),簡(jiǎn)書(shū)系信息發(fā)布平臺(tái),僅提供信息存儲(chǔ)服務(wù)。

相關(guān)閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容