在配置Tensorflow環境過程中，由於版本和驅動不對應，導致調試時一直報錯，請后續者在配置前，務必參看CUDA和CUDNN與Tensorflow的對應關系后再動手，避免不必要的時間浪費。本文系統環境是Ubuntu18.04,顯卡是GeForce GTX960M。鏈接2在較新的硬件安裝時用，可大輻節省時間。
本文參考鏈接1：https://blog.csdn.net/BigData_Mining/article/details/99670642
本文參考鏈接2：https://www.cnblogs.com/ManWingloeng/p/11489122.html
一、安裝N卡驅動：
1，查看N卡信息。
1 nvidia-smi
2，刪除舊版N卡驅動。
1 sudo apt-get remove nvidia-* 2 sudo apt-get autoremove
3，獲取系統推薦驅動（括號帶recommended的），並安裝該驅動，我的是440。
1 ubuntu-drivers devices
1 sudo apt-get install nvidia-settings nvidia-driver-440 nvidia-prime
4，重啟。
1 sudo reboot
二、安裝CUDA10.1及CUDNN7.6.1
1，查看適合本機的CUDA及CUDNN版本。
1 cat /usr/local/cuda/version.txt
1 cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
2，安裝CUDA。在http://developer.nvidia.com/cuda-downloads下載安裝deb包安裝，或直接通過其下介紹的Base Installer方式安裝，我這邊選了官網的方式（注意：下載時需要注冊一個NVIDIA賬號，才可繼續）。
1 wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu1804/x86_64/cuda-ubuntu1804.pin 2 sudo mv cuda-ubuntu1804.pin /etc/apt/preferences.d/cuda-repository-pin-600 3 wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb 4 sudo dpkg -i cuda-repo-ubuntu1804-10-1-local-10.1.243-418.87.00_1.0-1_amd64.deb 5 sudo apt-key add /var/cuda-repo-10-1-local-10.1.243-418.87.00/7fa2af80.pub 6 sudo apt-get update 7 sudo apt-get -y install cuda
3，安裝CUDNN。通過https://developer.nvidia.com/rdp/cudnn-archive下載對應的CUDNN壓縮文件，解壓獲得文件夾cuda，將其整個復制到系統主目錄home下。
4，設置環境變量（設置前要安裝好Vim)。
1 vi ~/.bashrc
輸入如上命令打開bashrc，按字母I鍵進行輸入模式，在最后面加入如下內容，按ESC鍵后，按shft+:，輸入wq保存並退出。
1 export PATH="/usr/local/cuda/bin:$PATH" 2 export LD_LIBRARY_PATH="/usr/lcoal/cuda/lib64:$LD_LIBRARY_PATH"
5，使設置生效。
1 source ~/.bashrc
三、通過pip安裝Tensorflow-gpu。
1 sudo pip install tensorflow-gpu