【Linux】Ubuntu 18.04 新机安装显卡驱动+CUDA+CuDNN 小记

最新推荐文章于 2024-02-07 18:20:18 发布

ZZ100861122

最新推荐文章于 2024-02-07 18:20:18 发布

阅读量283

点赞数

原文链接：https://blog.csdn.net/qq_39478403/article/details/109378705?utm_medium=distribute.pc_feed.none-task-blog-personrec_tag-3.nonecase&depth_1-utm_source=distribute.pc_feed.none-task-blog-personrec_tag-3.nonecase&request_id=5fa31d458478d54024af3afe

版权

一、安装显卡驱动

1.1 下载显卡驱动安装包

1.2 更换下载源

1.3 安装依赖

1.4 屏蔽开源驱动 nouveau

1.5 关闭图形界面

1.6 安装驱动

1.7 检查是否安装成功

1.8 重启 X-window 服务

二、CUDA 安装

2.1 安装

2.2 测试是否安装成功

三、cuDNN 安装

友情提示：仅供参考 —— 特殊情况应特殊分析

一、安装显卡驱动
1.1 下载显卡驱动安装包

官网链接：https://www.nvidia.cn/Download/index.aspx?lang=cn

手动查找适用于自己的 NVIDIA GPU 的驱动程序

然后，点击搜索，再点击下载：

1.2 更换下载源

更换国外/国内官方源为国内其他源 (如阿里源/清华源等)，避免下载速度过慢。打开 Terminal，依次输入以下命令：

备份 /etc/apt/sources.list 文件

mv /etc/apt/sources.list /etc/apt/sourses.list.backup

用 gedit 修改源列表内容

sudo gedit /etc/apt/sources.list

删除原内容，添加以下内容 (阿里源/清华源)

# 阿里云源
deb http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
# 源碼
deb-src http://mirrors.aliyun.com/ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb-src http://mirrors.aliyun.com/ubuntu/ bionic-proposed main restricted universe multiverse
# 清华大学源
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse
# 源碼
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic main restricted universe multiverse
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-security main restricted universe multiverse
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-updates main restricted universe multiverse
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-backports main restricted universe multiverse
##測試版源
deb-src http://mirrors.tuna.tsinghua.edu.cn/ubuntu/ bionic-proposed main restricted universe multiverse

保存并关闭 gedit 文本编辑界面

更新软件列表

sudo apt-get update

更新软件

sudo apt-get upgrade

1.3 安装依赖

分别执行两条命令安装安装 NVIDIA 显卡依赖软件

sudo dpkg --add-architecture i386
sudo apt install build-essential libc6:i386

1.4 屏蔽开源驱动 nouveau

终端输入

sudo gedit /etc/modprobe.d/blacklist.conf

在文本最后添加以下两行内容后，保存并关闭（用于禁用 nouveau 第三方驱动，之后也不需要改回来）

blacklist nouveau
options nouveau modeset=0

依次执行以下命令，使禁用生效并重启系统

sudo update-initramfs -u
sudo reboot

重启后，输入以下命令验证是否生效

lsmod | grep nouveau

若没有输出，则禁用生效
1.5 关闭图形界面

按 Ctrl-Alt+F1 进入命令行界面 (还有其它进入方式，比如 Ctrl-Alt+F1 ~ F7)，输入用户名和密码登录

输入以下命令，禁用 X-Window 服务

sudo service lightdm stop # 这会关闭图形界面

注意：我的电脑上没有安装 lightdm，图形界面是 GTK3 的，所以如果提示 unit lightdm.service not loaded

则需要先安装 lightdm

sudo apt-get install lightdm

安装完毕后跳出一个界面 (有 GTK3 和 lightdm 两个选项)，选择 lightdm

此时，再执行 sudo service lightdm stop 以禁用 X-Window 服务

（若安装过其他版本或其他方式安装过驱动执行此项，应先卸载原有驱动，以下是其中一种方式）

sudo apt-get remove nvidia-*

1.6 安装驱动

如果驱动文件默认下载在 Downloads 目录下，则先进入该目录

cd Downloads/

给驱动 run 文件赋予执行权限 (注意驱动文件名要与自己下载的相符)

sudo chmod a+x NVIDIA-Linux-x86_64-455.23.04.run

安装驱动 (注意驱动文件名要与自己下载的相符)

sudo ./NVIDIA-Linux-x86_64-455.23.04.run –no-opengl-files -no-x-check -no-nouveau-check

安装过程中都点同意即可，如果遇到下面的提示，在按键盘左、右键调整选项后回车继续安装：

The distribution-provided pre-install script failed! Are you sure you want to continue? -> CONTINUE INSTALLATION

Nvidia's 32-bit compatibility libraries? -> No

Would you like to run the nvidia-xconfigutility to automatically update your x configuration so that the NVIDIA x driver will be used when you restart x? Any pre-existing x confile will be backed up -> YES

注意，上述最重要的一步问你是否使用 nv 的 xconfig 文件，一定要选 yes，否则在启动 x-window 时不会使用 nv 驱动！
1.7 检查是否安装成功

在命令行界面 / 终端均可检查

nvidia-smi # 如果出现GPU列表，则驱动安装成功

输出类似以下的信息，即表示正常安装成功

Fri Oct 30 13:34:38 2020       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 455.23.04    Driver Version: 455.23.04    CUDA Version: 11.1     |
|-------------------------------+----------------------+----------------------+
| 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  GeForce RTX 3090    Off  | 00000000:01:00.0  On |                  N/A |
|  0%   31C    P8    22W / 350W |     52MiB / 24265MiB |      1%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1036      G   /usr/lib/xorg/Xorg                 50MiB |
+-----------------------------------------------------------------------------+

1.8 重启 X-window 服务

sudo service lightdm start

然后按 Ctrl-Alt+F7 即可恢复到图形界面。

安装完成！
二、CUDA 安装
2.1 安装

下载后，切换到文件目录下，执行安装命令 (文件名要与自己下载的相符)：

sudo sh cuda_11.1.0_455.23.05_linus.run

然后，出现一个安装选项界面，大概像下面这个样子：

因为已经提前手动安装了显卡驱动，所以此处把显卡驱动的安装选项取消掉 (用键盘移动到 [X] Driver 按回车取消选中 [ ] Driver)：

选择 Install 然后回车，出现如下界面，表示安装完成 (不必担心警告，因为那是我们没有选择安装显卡驱动选项而出现的，忽略即可)

然后，添加环境变量，编辑.bashrc配置文件：

sudo gedit ~/.bashrc

在文件最后，写入下面几句话，注意 cuda 版本要和自己的相符：

export  PATH=/usr/local/cuda-11.1/bin:$PATH
export  LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64$LD_LIBRARY_PATH

保存并退出

source ~/.bashrc

2.2 测试是否安装成功

上面的选项选择安装了 CUDA 例子，运行其中一个来测试是否安装成功：

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery

结果如下就表明成功啦

三、cuDNN 安装

需要先在 cuDNN 官方注册、登陆再下载对应 CUDA 版本的 cuDNN，然后解压。此处略过了该步骤，假设已经解压完毕，则生成一个 cuda 文件夹，执行如下命令拷贝文件：

$ 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*

输入以下命令，验证是否安装成功

cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

如安装成功，会出现类似如下的界面

至此安装完成！

ZZ100861122

关注

0
点赞
踩
2

收藏

觉得还不错? 一键收藏
0
评论
【Linux】Ubuntu 18.04 新机安装显卡驱动+CUDA+CuDNN 小记

目录一、安装显卡驱动1.1 下载显卡驱动安装包1.2 更换下载源1.3 安装依赖1.4 屏蔽开源驱动 nouveau1.5 关闭图形界面1.6 安装驱动1.7 检查是否安装成功1.8 重启 X-window 服务二、CUDA 安装2.1 安装2.2 测试是否安装成功三、cuDNN 安装友情提示：仅供参考 —— 特殊情况应特殊分析一、安装显卡驱动1.1 下载显卡驱动安装包官网链接：https://www.nvidia.cn/Download/index.aspx?lang=
复制链接

扫一扫