本文的安裝環(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)大