1.安裝NVIDIA驅動
首先去官網（http://www.nvidia.cn/Download/index.aspx?lang=cn）查找適配自己電腦GPU的驅動，我的電腦驅動版本如下：
執行如下語句安裝NVIDIA驅動
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384 //驅動版本不同，對應不同的數字
sudo apt-get install mesa-commoon-dev //本句及下一句命令有的電腦需要，有的電腦不需要，根據實際情況輸入命令
sudo pat-get install freeglut3-dev
執行完上述命令後若無問題，重啓，若有問題，單獨查資料解決
sudo reboot
重啓後輸入：
sudo nvidia-smi
如果顯示GPU列表，則證明驅動安裝成功了，另外也可以通過如下命令：
sudo nvidia-settings
查看GPU信息，顯示界面如下：
2.安裝CUDA9.0
首先去官網下載CUDA 9.0，這裏因爲寫該博客時NVIDIA網站進不去了，就截了一個圖，下載的是.run文件
下載完CUDA 9.0之後執行如下語句，運行.run文件
sudo sh cuda_9.0.176_384.81_linux.run
單擊回車，一路往下運行，直到提示“是否爲NVIDIA安裝驅動nvidia-384？”，選擇否，因爲已經安裝好驅動程序了，其他的全都是默認，不過要記住安裝位置，默認是安裝在/usr/local/cuda文件夾下。
配置環境變量，運行如下命令打開profile文件
sudo gedit /etc/profile
打開文件後在文件末尾添加路徑，也就是安裝目錄，命令如下：
export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64$LD_LIBRARY_PATH
保存，然後重啓電腦
sudo reboot
3.測試CUDA的Samples例子
cd /usr/local/cuda-9.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
如果顯示的是關於GPU的信息，則說明安裝成功了。
4. 然後解壓壓縮包：cudnn-9.0-linux-x64-v7.tgz
然後把cuDNN裏面的一些文件拷貝到剛安裝好的cuda-9.0的安裝目錄下，並且設置權限。命令爲：
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
---------------------
作者：天澤28
來源：CSDN
原文：https://blog.csdn.net/u012328159/article/details/80959454
版權聲明：本文爲博主原創文章，轉載請附上博文鏈接！