Ubuntu20.04 安装NVIDIA驱动+ CUDA11

0 / 916

安装NVIDIA驱动

这里用手动run文件安装NVIDIA驱动。

一. 下载驱动
查看显卡型号,输入以下代码。

lspci | grep -i nvidia

在NVIDIA官网选择显卡对应版本驱动下载
这是官网地址

选择完点击搜索,下载所需版本的驱动(我下的最新版本)。

二. 安装驱动
删除已经安装的驱动

sudo apt-get remove --purge nvidia-*

禁用NOUVEAU
输入以下命令,查看lsmod是否在使用

lsmod | grep nouveau

如有输出,则在使用,没有输出继续往下走。

按CTRL+AIT+F1进入字符界面,我按的F3也行。
输入命令关闭图形界面

sudo service lightdm stop

输入完我这里显示service not loaded. 所以就没管他继续往下做(如果不是这样在安装完需要输入sudo service lightdm start)。

赋予run文件执行权限

sudo apt-get install gcc-7 g++-7
sudo update-alternatives --install /usr/bin/gcc gcc /usr/bin/gcc-7 100
sudo chmod a+x NVIDIA-Linux-x86_64-455.45.01.run

# 安装驱动
sudo ./NVIDIA-Linux-x86_64-455.45.01.run -no-x-check -no-nouveau-check -no-opengl-files

-no-x-check -no-nouveau-check -no-opengl-files 一定要加上

安装过程中的选项为y、y、n。
安装完重启,输入nvidia-smi,显示以下信息则成功。

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 515.67       Driver Version: 517.00       CUDA Version: 11.7     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ...  On   | 00000000:01:00.0 Off |                  N/A |
| N/A   47C    P0    17W /  N/A |      0MiB /  6144MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|  No running processes found                                                 |
+-----------------------------------------------------------------------------+

安装CUDA11

一. 下载CUDA11
首先去官网下载对应的CUDA:https://developer.nvidia.com/cuda-downloads

二. 安装CUDA
输入sudo sh cuda_11.0.1_450.36.06_linux.run
安装过程选项:

accept
driver这里不用选,如果是y or n 选 n,如果是X,在X那里输入回车取消选择。(第一次我选了y结果安装失败,又手动删了驱动重新装一遍)
toolkit必选
后面的随意,我默认都选了

三. 配置环境变量
输入gedit ~/.bashrc,在最后添加两行:

export PATH=/usr/local/cuda-10.2/bin{PATH:+:{PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64{LD_LIBRARY_PATH:+:{LD_LIBRARY_PATH}}

目录里的cuda路径记得修改。
保存,退出terminal,重新开一个后输入source ~/.bashrc

安装完成,输入nvcc -V 查看cuda版本。