因為必須關閉桌面窗口, 建議另外一台電腦ssh連接操作
1. 卸載舊版本並關閉圖形界面
sudo apt-get remove --purge nvidia-\*
sudo service lightdm stop
sudo init 3
2. 安裝對應版本號驅動
sudo chmod +x cuda_8.0.61_375.26_linux.run sudo sh cuda_8.0.61_375.26_linux.run --tmpdir=/tmp --override
3. 安裝cudnn5.1
tar -xzvf cudnn-8.0-linux-x64-v5.1.tgz sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64 sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn* sudo ln -sf libcudnn.so.5.1.10 libcudnn.so.5 sudo ln -sf libcudnn.so.5 libcudnn.so sudo ldconfig
4. 查看安裝版本號並重啟圖形界面
nvcc --version
sudo service lightdm start
5. 然后,切記在系統設置,軟件更新中設置一下停止自動更新.
2. cuda下載及安裝
sudo apt-get install bazel https://github.com/tensorflow/tensorflow.git ./configure make sudo make install