Ubuntu16.04 安裝 CUDA8.0 + cudnn5.1 + TensorFlow(GPU) 詳細過程

下面介紹如何在Ubuntu16.04 中安裝GPU驅動，我電腦配置的顯卡是 TITAN X 。所以以下安裝將以 TITAN X 爲例介紹。

（1）首先要安裝ubuntu16.04TLS系統在電腦中，我的電腦安裝的是雙系統，我沒有試過在虛擬機中安裝，不知道以下方法能不能在虛擬機中安裝。

（2）安裝顯卡驅動，在系統設置-->軟件和更新-->附加驅動-->選擇需要的驅動         (手動設置)

（3）安裝 cuda-8.0  兩種方法：

            a ：下載安裝包：cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb

                   在下載路徑中執行如下命令安裝：

                    sudo dpkg -i cuda-repo-ubuntu1604-8-0-local_8.0.44-1_amd64.deb

                    sudo apt-get update

                    sudo apt-get install cuda

                    （建議使用）

             b ： 下載安裝包：cuda_8.0.44_linux.run

                    在下載路徑中執行如下命令安裝：

                    chmod u+x cuda_8.0.44_linux.run

                    sudo ./cuda_8.0.44_linux.run

                   注意：安裝過程中不需要再安裝顯卡驅動

（4）安裝cudnn--> cuda

           下載包：cudnn-8.0-linux-x64-v5.1.tgz 解壓後得到 cuda 文件夾

             在cuda 目錄下執行如下命令：

                      sudo cp cuda/include/cudnn.h /usr/local/cuda-8.0/include/

                      sudo cp cuda/lib64/libcudnn* /usr/local/cuda-8.0/lib64/

（5）配置環境變量：vi .bashrc

            添加如下路徑：

               export LD_LIBRARY_PATH=/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
               export CUDA_HOME=/usr/local/cuda-8.0
               export PATH=/usr/local/cuda-8.0/bin:$PATH 

         保存退出   

            執行：source .bashrc  使配置文件生效

（6）安裝 tensorflow(GPU)

             sudo pip install tensorflow-gpu  將會安裝tensorflow最新的版本

         

          安裝完成後會安裝一些其他的一些依賴庫。

（7）驗證 tensorflow -gpu 是否安裝成功。