1、安裝顯卡驅動

到英偉達官網(wǎng)下載對應的顯卡驅動并安裝

2、下載cuda8.0的deb包

下載地址
https://developer.nvidia.com/cuda-80-ga2-download-archive
安裝指南https://developer.download.nvidia.com/compute/cuda/8.0/secure/Prod2/docs/sidebar/CUDA_Quick_Start_Guide.pdf?lA8eVJXIeTzh4z8j5EpvWudlP9GfoauMt2RUNGI2wU-ZQ7EJdSk5FjiXFt_teTE6xPLauRUPk8QEfsHsJ-g1AzPwe4Ho8u4ETGz0ImOl8cEk9Ionfj8sW36yNBMn-PCOKLcXMIGacd_otdCir1jEngA3ZdDooRdFyPU0Od76dpBPhlbh
設置環(huán)境變量
gedit ~/.bashrc
最后添加下面兩行
export PATH=/usr/local/cuda-8.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

3、下載安裝cudnn6.0

tar -xvf cudnn-8.0-linux-x64-v5.1.tgz
cd cuda
sudo cp ./include/cudnn.h /usr/local/cuda-8.0/include
sudo cp -a ./lib64/libcudnn* /usr/local/cuda-8.0/lib64

4、安裝tensorflow

pip install tensorflow-gpu==1.3