安裝nvidia - HackMD

--- tags: Linux --- # 安裝nvidia <style> .red{color: red;} </style> 使用的是ubuntu18.04、GTX1050ti ## 步驟一安裝驅動 [參考連結](https://blog.csdn.net/opencv_fjc/article/details/106094594) 1. 確保本機中沒有未卸載乾淨的NVIDIA驅動 ``` $ sudo apt-get remove --purge nvidia* ``` 2. Ubuntu系統默認安裝好是使用一個開源驅動:```nouveau```，但在安裝官方的驅動時須先禁用這個開源驅動 ``` $ sudo bash -c "echo blacklist nouveau > /etc/modprobe.d/blacklist-nvidia-nouveau.conf" $ sudo bash -c "echo options nouveau modeset=0 >> /etc/modprobe.d/blacklist-nvidia-nouveau.conf" ``` 執行完上面兩個指令後，可以使用指令 ```$ cat /etc/modprobe.d/blacklist-nvidia-nouveau.conf``` 查看是否成功禁用開源驅動(下面圖片顯示禁用成功畫面) : ![](https://i.imgur.com/50OfZOF.png) 3. 添加PPA源 ``` $ sudo add-apt-repository ppa:graphics-drivers/ppa $ sudo apt-get update ``` 4. 安裝驅動有兩種方法(**二選一**) : * 使用ubuntu自動安裝命令進行安裝 (我用這個) ``` $ sudo ubuntu-drivers autoinstall ``` * 安裝指定版本的驅動 ``` ## 查看推薦的驅動版本指令 $ ubuntu-drivers devices ``` ![](https://i.imgur.com/nnnqLtb.png) 從中選擇一個 ``` ## 假如我選擇 nvidia-driver-470 $ sudo apt-get install nvidia-driver-470 ``` 5. 查看安裝結果使用 ```nvidia-smi``` 查看，如果有下面的畫面代表成功 ![](https://i.imgur.com/WtNOzI7.png) :::danger 假如沒有，呵呵，加油~ 像我有出現一個錯誤顯示 : NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver. Make sure that the latest NVIDIA driver is installed and running. [解決](https://zhuanlan.zhihu.com/p/89714824) : 查看安裝的nvidia驅動版本號 ``` $ ls /usr/src | grep nvidia ``` ![](https://i.imgur.com/70C5idb.png) ``` $ sudo apt install dkms $ sudo dkms install -m nvidia -v 495.44 $ reboot ``` ::: :::danger * 假如上面還是有問題，可以試著抓看看有沒有顯卡 : ``` $ lspci | grep 'VGA' ``` * 確認 BIOS 裡面的 security boot是否為disable的狀態 ![](https://i.imgur.com/wfog5c9.png) ::: --- ## 步驟二安裝cuda [參考連結](https://www.twblogs.net/a/5d3fee57bd9eee51fbf95548) 我安裝的是[cuda10.1](https://developer.nvidia.com/cuda-10.1-download-archive-base?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=1804&target_type=runfilelocal) ![](https://i.imgur.com/6hU4Cxy.png) 1. 下載、解壓縮 * 先右邊的download下載 * 下載完輸入藍底部分的指令 * 輸入完指令會給你選擇要安裝什麼，把第一個driver去掉，因為第一步驟安裝過了 2. 配置環境變量 在```~/.bashrc```尾部配置環境變量 ``` # ~/.bashrc裡面 export PATH=/usr/local/cuda-10.1/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH ``` 保存退出後輸入下面指令及生效 ``` $ source ~/.bashrc ``` 3. 設置環境變量和動態鏈接庫 ``` $ sudo vim /etc/profile ``` 在文件尾端加入 : ``` export PATH=/usr/local/cuda-10.1/bin:$PATH ``` 4. 創建鏈接文件 ``` $ sudo vim /etc/ld.so.conf.d/cuda.conf ``` 在打開的文件中添加下面的語句(10.1改成自己的版本號) ``` /usr/local/cuda-10.1/lib64 ``` 保存退出，然後執行 ``` $ sudo ldconfig ``` 5. 測試cuda是否安裝成功 ``` $ cd ~/NVIDIA_CUDA-10.1_Samples $ sudo make all -j8 $ cd bin/x86_64/linux/release $ ./deviceQuery ``` 如果cuda安裝成功，會顯示 Result = PASS ![](https://i.imgur.com/4n0fncX.png) --- ## 步驟三安裝cudnn [參考連結](https://www.twblogs.net/a/5d3fee57bd9eee51fbf95548) [下載地址](https://developer.nvidia.com/rdp/cudnn-archive) ![](https://i.imgur.com/65GMF4H.png) * 安裝下載後，使用下面指令(其實就是安裝下載的三個東西) ``` $ sudo dpkg -i libcudnn7_7.6.1.34-1+cuda10.1_amd64.deb $ sudo dpkg -i libcudnn7-dev_7.6.1.34-1+cuda10.1_amd64.deb $ sudo dpkg -i libcudnn7-doc_7.6.1.34-1+cuda10.1_amd64.deb ``` * 測試cuDNN 複製cuDNN sample到一個個目錄下，這裡複製到HOME ``` $ cp -r /usr/src/cudnn_samples_v7 /$HOME ``` 進入 ``` $ cd $HOME/cudnn_samples_v7/mnistCUDNN/ ``` 編譯 mnistCUDNN sample ``` $ make clean && make all –j8 ``` 運行 ``` $ ./mnistCUDNN ``` 如果出現 Test passed! 代表cuDNN已成功安裝 ![](https://i.imgur.com/62ki5r3.png) :::danger 查看cudnn版本 ``` $ dpkg -l | grep cudnn #查看已安装版本 ``` 刪除版本 ``` $ sudo dpkg -r libcudnn7 # 删除版本 ``` :::