該方案爲筆者嘗試安裝正確後的經驗總結,參考文檔:
- Ubuntu安裝和卸載CUDA和CUDNN
- Ubuntu安裝gcc g++ 7
- NVIDIA CUDA Installation Guide for Linux
- Ubuntu 18.04 gcc降級爲4.8版本
- ubuntu18.04 + CUDA10 + CUDNN7 deb安裝
一、安裝包準備
首先下載所需安裝包,驅動、CUDA-10.2、CUDNN7_7
1.下載驅動
驅動官網下載地址:https://www.nvidia.cn/Download/index.aspx?lang=cn ,根據自己顯卡的情況下載對應版本的顯卡驅動,比如筆者的顯卡是RTX2060 SUPER
下載成功後得到一個後綴爲.run的安裝包:
NVIDIA-Linux-x86_64-440.36.run
2.下載CUDA
我們可以在官網:CUDA10下載頁面根據相應版本進行下載,筆者下載的是:
cuda_10.2.89_440.33.01_linux.run
3. 下載CUDNN
進入CUDNN官網,然後點擊Download開始選擇下載版本,當然在下載之前還有登錄,選擇版本界面如下,我們選擇cuDNN Library for Linux,筆者下載的是cuDNN Runtime Library for Ubuntu18.04 (Deb)的文件,這個比壓縮包安裝的方式更快捷:
libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
二、安裝顯卡驅動
1. 禁用nouveau驅動
sudo vim /etc/modprobe.d/blacklist.conf
在文本後添加:
blacklist nouveau
options nouveau modeset=0
然後執行:
sudo update-initramfs -u
重啓後,執行以下命令,如果沒有屏幕輸出,說明禁用nouveau成功:
lsmod | grep nouveau
2. 安裝新驅動
切換到命令行終端:
Ctrl-Alt+F3
執行以下命令禁用X-Window服務,否則無法安裝顯卡驅動:
sudo service lightdm stop
安裝新驅動:
sudo chmod +x NVIDIA-Linux-x86_64-440.36.run
sudo ./NVIDIA-Linux-x86_64-440.36.run
執行以下命令啓動X-Window服務:
sudo service lightdm start
最後執行重啓命令,重啓系統即可:
reboot
注意: 如果系統重啓之後出現重複登錄的情況,多數情況下都是安裝了錯誤版本的顯卡驅動。需要下載對應本身機器安裝的顯卡版本。
三、安裝CUDA-10.2和CUDNN-7_7
由於CUDA-10.2支持的gcc版本爲7.3,而ubuntu-19.10自帶的gcc/g++版本爲9.2,故需要安裝gcc-7.3。
1. 安裝gcc和g++,從9.2降級到7.3或者7.4
sudo add-apt-repository ppa:ubuntu-toolchain-r/test
sudo apt-get update
sudo apt-get install gcc-7
sudo apt-get install g++-7
把gcc g++ 的連接指向gcc-7 g+±7
$ cd /usr/bin
$ sudo rm gcc
$ sudo ln -s gcc-7 gcc
$ sudo rm g++
$ sudo ln -s g++-7 g++
2. 安裝CUDA
chmod +x cuda_10.2.89_440.33.01_linux.run
sudo ./cuda_10.2.89_440.33.01_linux.run
注意:在安裝時,會彈出安裝組件的界面,將第一個選項驅動安裝取消掉,否則會安裝失敗
安裝完成之後,可以配置他們的環境變量,在vim ~/.bashrc的最後加上以下配置信息:
export CUDA_HOME=/usr/local/cuda-10.2
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64
export PATH=${CUDA_HOME}/bin:${PATH}
最後使用命令source ~/.bashrc使它生效。
可以使用命令nvcc -V查看安裝的版本信息。
測試是否安裝成功,執行以下幾條命令:
cd /usr/local/cuda-10.0/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
正常情況下會輸出GPU信息。
3. 安裝CUDNN
sudo dpkg -i libcudnn7_7.6.5.32-1+cuda10.2_amd64.deb
安裝完成。