NVIDIA驱动安装（arm64平台可用）

一、下载驱动

1、确定显卡型号

使用lspci指令，查看自己的显卡型号

xxxx:~$ lspci | grep NVIDIA
10:00.0 3D controller: NVIDIA Corporation TU104GL [Tesla T4] (rev a1)

可以看到，我的显卡是Tesla T4.

2、去官网下载驱动

访问nv官网，驱动下载，根据自己的显卡型号搜索对应的驱动

注意不要选错操作系统了，我的是ARM平台的，选择的aarch64，然后点击搜索。

可以看到驱动版本等信息，还可以从支持的产品列表里看到的我的显卡，Tesla T4。下载下来。

二、禁用Nouveau驱动

1、配置文件禁用

nouveau是linux内核中自带的nv驱动，官方驱动会跟它有冲突，所以需要禁用掉。
在 /etc/modprobe.d/blacklist-nouveau.conf（如果没有该文件就创建一个）下添加以下两行并保存。

blacklist nouveau
options nouveau modeset=0

2、重做initramfs镜像

这个在不同操作系统上的操作是不一样的。

1）centos

dracut --force
cp /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
dracut /boot/initramfs-$(uname -r).img $(uname -r)

并重启服务器

2）ubuntu

update-initramfs -u

并重启服务器

3、检验

lsmod | grep nouveau

如果无输出，则表示禁用成功。

三、预安装组件（centos）

ubuntu系统可以忽略此步骤

yum install gcc kernel-devel kernel-headers

yum -y install centos-release-scl
yum -y install devtoolset-8-gcc devtoolset-8-gcc-c++ devtoolset-8-binutils
echo "source /opt/rh/devtoolset-8/enable" >>/etc/profile
source /etc/profile

四、安装驱动

添加驱动可执行权限

chmod +x NVIDIA-Linux-aarch64-450.80.02.run

安装

./NVIDIA-Linux-aarch64-450.80.02.run

安装时基本上都选默认就可以。

五、验证安装成功

nvidia-smi

输出类似一下的信息，就是安装成功了。