更新时间:2021-07-13 18:10:38
Linux安装Nvidia Tesla驱动有两种方式。
run程序安装
适用于所有Linux发行版。
驱动包安装
如deb包,rpm包。适用于常见的Linux发行版本。
安装过程中,驱动会编译kernel module,要求操作系统安装gcc和kernel-devel等依赖包。
以下安装步骤以在Centos7.4操作系统安装为例。
打开NVIDIA驱动下载链接:http://www.nvidia.com/Download/Find.aspx。
选择操作系统和安装包。如,使用K20卡,选择要下载的版本。
注:VGPUK10X类型选择tesla K10驱动,VGPUK20X类型选择tesla K20m驱动。对于CUDA Toolkit版本,如无特殊要求,建议使用9.1及以上版本。
使用run程序安装,操作系统选择框选择“Linux 64-bit”,搜索驱动。
选择所需的驱动版本,页面跳转后点击DOWNLOAD。
跳转到下载页面后,点击AGREE & DOWNLOAD进行下载,下载到本地后再导入到云服务器中。或者右击AGREE & DOWNLOAD复制链接地址,在云服务器内直接wget下载。
如,
wget https://us.download.nvidia.com/tesla/440.64.00/NVIDIA-Linux-x86_64-440.64.00.run 。
安装当前系统的kernel-devel和kernel-headers。由于当前系统的内核版本对应的kernel-devel和kernel-headers可能在yum源上下载不了,您可以在网上搜索对应的rpm包下载,并在操作系统内安装。参考链接:http://rpm.pbone.net/。
您也可以直接用yum安装最新的kernel,kernel-devel和kernel-headers。
编译驱动要求kernel和kernel-devel版本一致,完成上述安装后,通过sudo rpm -qa |grep kernel查看是否安装成功。
安装gcc包。gcc依赖kernel-headers包,在上一步已经完成安装了。这里直接yum安装gcc。
yum install gcc -y
vi /etc/modprobe.d/dist-blacklist.conf
# generated by nvidia-installer
blacklist nouveau
options nouveau modeset=0
完成后需要重启云服务器,使其重新加载。
chmod +x NVIDIA-Linux-x86_64-440.64.00.run
./ NVIDIA-Linux-x86_64-440.64.00.run
安装完成后,使用nvidia-smi
命令验证是否已经安装成功。
根据您的GPU云服务器选择对应的操作系统,如CentOS7.4选择Linux 64-bit RHEL7,Ubuntu16.04选择Linux 64-bit Ubuntu 16.04。
选择其中一个驱动版本进行点击,会跳转到下载页面。点击DOWNLOAD跳转后,点击AGREE & DOWNLOAD进行下载,将rpm包导入到GPU云服务器中(或者右键复制链接,在云服务器内通过wget进行下载)。
准备好安装包后。点击ADDITIONAL INFORMATION查看安装步骤,下面以CentOS7.3和Ubuntu16.04为例说明。
注意1:CentOS安装cuda-driver前需要先安装与内核版本一致kernel-devel包。由于当前系统的内核版本对应的kernel-devel可能在yum源上下载不了,您可以在网上搜索对应的rpm包下载(通过
uname -r
查看版本号),并在操作系统内安装,参考链接:http://rpm.pbone.net。
您也可以直接用yum安装最新的kernel,kernel-devel,安装后需要重启。
注意2:执行yum install cuda-drivers时,可能会报缺少dkms包的错误。这时候需要先安装dkms。
yum install epel-release -y
yum install dkms -y
注意1:执行
dpkg -i nvidia-diag-driver-local-repo-ubuntu1604-390.46_1.0-1_amd64.deb
命令后,会提示安装public CUDA GPG key,执行其提示的安装命令即可。如下图所示。
注意2:在安装过程中,如果出现以下错误,一般是apt源不可用,可参考Linux更新yum源或apt源进行更换。
注意3:apt-get install cuda-drivers安装驱动时,会默认安装图形界面。安装完成后,如果您通过控制台远程连接到GPU云服务器上,会默认进入图形界面。通过快捷键ctrl+alt+F2可以切换到dos命令行界面。如果您不需要图形界面,可以自行卸载。