CentOS7安装nvidia驱动并在docker容器中使用GPU

最新推荐文章于 2024-10-17 20:02:59 发布

原创

最新推荐文章于 2024-10-17 20:02:59 发布 · 3.3k 阅读

标签

#docker #cuda #gpu

本文介绍如何在CentOS上安装NVIDIA GPU驱动及CUDA工具包，包括调整内核版本、禁用默认驱动、安装NVIDIA驱动和CUDA，以及在Docker环境中配置NVIDIA容器运行时。

1.install NVIDIA Driver

a. 需要先确保kernel-devel和kernel-headers包的版本和当前kernel版本一致

如果不一致需要卸载高版本的kernel-devel包或者升级kernel，这里建议降低kernel-devel的版本。
如下所示通过yum安装的kernel-devel是3.10.0-1160版本的，然而kernel是3.10.0-1127版本的

###安装 kernel-devel
[jenkins_user@chemical_122 gpu_related]$ sudo yum install epel-release -y
###这里gcc和gcc-c++的版本比较低，对后面编译有影响，如果要做流水线编译环境，得编译安装高版本gcc，一般部署环境不需要
[jenkins_user@chemical_122 gpu_related]$ sudo yum install -y net-tools telnet autoconf libtool make gcc gcc-c++ kernel-devel

[root@chemical_122 ~]# rpm -qa | grep kernel-
kernel-tools-libs-3.10.0-1127.el7.x86_64
kernel-devel-3.10.0-1160.11.1.el7.x86_64
kernel-headers-3.10.0-1160.11.1.el7.x86_64
kernel-tools-3.10.0-1127.el7.x86_64
kernel-3.10.0-1127.el7.x86_64
[root@chemical_122 ~]# uname -a
Linux chemical_122 3.10.0-1127.el7.x86_64 #1 SMP Tue Mar 31 23:36:51 UTC 2020 x86_64 x86_64 x86_64 GNU/Linux

###下载对应版本的kernel-devel的rpm包，并安装 
[root@chemical_122 ~]# yum remove -y kernel-devel
[root@chemical_122 software]# wget https://mirrors.tuna.tsinghua.edu.cn/centos-vault/7.8.2003/os/x86_64/Packages/kernel-devel-3.10.0-1127.el7.x86_64.rpm
[root@chemical_122 software]# wget https://mirrors.tuna.tsinghua.edu.cn/centos-vault/7.8.2003/os/x86_64/Packages/kernel-headers-3.10.0-1127.el7.x86_64.rpm
[root@chemical_122 software]# rpm -ivh kernel-devel-3.10.0-1127.el7.x86_64.rpm
[root@chemical_122 software]# rpm -ivh kernel-headers-3.10.0-1127.el7.x86_64.rpm

b. 禁用默认的驱动并重启安装NVIDIA驱动

[jenkins_user@chemical_122 gpu_related]$ wget https://cn.download.nvidia.cn/XFree86/Linux-x86_64/430.14/NVIDIA-Linux-x86_64-430.14.run 
[jenkins_user@chemical_122 gpu_related]$ chmod ug+x ./NVIDIA-Linux-x86_64-430.14.run
[jenkins_user@chemical_122 gpu_related]$ sudo ./NVIDIA-Linux-x86_64-430.14.run
Verifying archive integrity... OK
Uncompressing NVIDIA Accelerated Graphics Driver for Linux-x86_64 430.14........................................................................................................................................................................................................................................