下载arm64版本驱动
axel -n 200 https://developer.download.nvidia.com/compute/cuda/11.8.0/local_installers/cuda_11.8.0_520.61.05_linux_sbsa.run
禁用nouveau
首次安装时,选择驱动一同安装,会自动生成文件/etc/modprobe.d/nvidia-installer-disable-nouveau.conf , 内容如下
# generated by nvidia-installer
blacklist nouveau
options nouveau modeset=0
重新生成initramfs
mkinitrd /boot/initramfs-4.19.90-23.8.v2101.ky10.aarch64.img 4.19.90-23.8.v2101.ky10.aarch64 --force
设置重启到字符界面
systemctl set-default multi-user.target
reboot
---------------------------------
重启后
安装依赖
yum install gcc kernel-devel kernel-headers
单独安装驱动
bash NVIDIA-Linux-aarch64-520.61.05.run --kernel-source-path=/usr/src/kernels/4.19.90-23.8.v2101.ky10.aarch64/
注:NVIDIA-Linux-aarch64-520.61.05.run可以在cuda_11.8.0_520.61.05_linux_sbsa.run运行阶段在/tmp目录下搜索到:find /tmp | grep NVIDIA
安装cuda
bash cuda_11.8.0_520.61.05_linux_sbsa.run
成果