ubuntu16.04 tensorflow cuda8.0 cudnn5.1

03-03

安裝顯卡驅動

系統設置->軟體更新->附加驅動->切換到最新的NVIDIA驅動即可。應用更改->重啟

再運行nvidia-smi來看看

gcc降版本

ubuntu的gcc編譯器是5.4.0，然而cuda8.0不支持5.0以上的編譯器，因此需要降級，把編譯器版本降到4.9：

在terminal中執行：

sudo apt-get install g++-4.9sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-4.9 20sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-5 10sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-4.9 20sudo update-alternatives --install /usr/bin/g++ g++ /usr/bin/g++-5 10sudo update-alternatives --install /usr/bin/cc cc /usr/bin/gcc 30sudo update-alternatives --set cc /usr/bin/gccsudo update-alternatives --install /usr/bin/c++ c++ /usr/bin/g++ 30sudo update-alternatives --set c++ /usr/bin/g++

安裝CUDA

sudo dpkg -i cuda-repo-ubuntu1604-8-0-local-ga2_8.0.61-1_amd64.deb

sudo apt-get update

sudo apt-get install cuda

安裝完畢後，再聲明一下環境變數，並將其寫入到 ~/.bashrc 的尾部:

export CUDA_HOME=/usr/local/cudaexport PATH=$PATH:$CUDA_HOME/binexport LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

保存退出，運行source ~/.bashrc

測試是否安裝成功

cd /usr/local/cuda/samples/1_Utilities/deviceQuery sudo make./deviceQuery

結果如下

安裝Cudnn

下載tar，解壓後會得到一個Cuda文件夾，複製到Cuda-8.0文件夾中

sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/includesudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda-8.0/lib64/libcudnn*

安裝TensorFlow

pip3 install tensorflow-gpu==1.4

安裝TensorFlow arise an error: you are using pip version 8.1.1,however version 9.0.1 is available