Rocky Linux9.1 下的nVidia驱动安装及CUDAtoolkit安装教程

张九幕

分子动力学是一种多维度、多角度研究蛋白质体系的计算方法。其中Gromacs是最常见和好用的实现工具之一。纯CPU版的Gromacs计算速度很慢，往往达不到我们的计算需求，因此需要安装GPU加速版的Gromacs。在Sobereva老师的Gromacs安装教程（GROMACS的安装方法（含全程视频演示） - 思想家公社的门口：量子化学·分子模拟·二次元 (sobereva.com)）中，给出详尽且贴心的指导。其中关于GPU加速的部分，要求安装者需要提前在自己的机子上安好CUDAtoolkit；曾经的笔者是一个连Linux命令都不怎么熟悉的小白，想要在Linux上安装好nVidia驱动及CUDAtoolkit，实在是有些困难，在笔者的不断尝试下，并参考欢乐多老师这篇博文（Rocky Linux 9.3系统安装及Gromacs , Gaussian , xtb , ORCA , SVM等程序安装 - 计算机使用与Linux交流 (Computer Usage and Linux) - 计算化学公社 (keinsci.com)），摸索出一套可行的安装方法，为以后有像笔者一样的小白踩坑，在这里将我的安装方法分享出来，供大家参考。如有不正确的地方，也请大家在评论区批评指正。

操作系统：Rokcy-linux 9.1

一、首先禁用nouveau
1.输入以下命令，对该文件进行修改：
vim /usr/lib/modprobe.d/dist-blacklist.conf
将blacklist nvidiafb一项删除或者顶格加“#”，使之成为注释。
在文件末尾加入：
blacklist nouveau
options nouveau modest=0
保存并退出（需要root权限）
2.输入以下命令（需要root权限）：
rmmod nouveau

重启系统后，输入：
lsmod | grep nouveau
发现没有弹出信息，则证明 nouveau以及成功禁用。
如果弹出一大堆信息，并且伴有nouveau字样，则说明没有禁用成功，请仔细检查前面步骤。

二、安装nVidia驱动
CUDAtoolkit本身自带驱动，但是笔者安装后会有图形界面不显示的问题。如果先安装驱动，再安装CUDAtoolkit，则没有这种问题。
到nVidia官网NVIDIA GeForce 驱动程序 - N 卡驱动 | NVIDIA，根据自身显卡型号搜索驱动，下载最新版即可。
用mobaxterm将驱动文件载入到机子中，通过右键点击文件，选中“Permission”，在“Execute”上单击打勾，赋予其执行权限。
输入以下命令，执行文件：
./NVIDIA-Linux-x86_64-555.42.02.run
（右侧文件名）
根据自身需求安装驱动，这一部分会有安装向导。

三、安装CUDA toolkit
到nVidia CUDA toolkit下载界面下载对应版本的CUDA toolkit，请注意，其版本不能超过当前驱动的版本，为避免混淆，最好使用同一版本。最好使用特殊方式进入这个网站下载，否则会很慢。
下载好之后，将文件通过mobaxterm拖入到机子中，同样的方式赋予执行权限，输入以下命令执行：
./cuda-Linux-x86_64-555.42.02.run
（右侧为文件名）
执行后，可以选择安装选项，“X”是表示选中的意思，而非不选中。选择CUDA toolkit安装即可。
设置环境变量：
vim ~/.bashrc
将以下两行命令加入：
export PATH="/usr/local/cuda-12.5/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-12.5/lib64:$LD_LIBRARY_PATH"

（注意要搞清楚自己的版本）
安装完毕后，可以通过nvcc -V查看，如果弹出自身CUDA toolkit的版本信息，则证明安装成功
输入nvidia-smi也会有信息弹出

此时，对于mobaxterm远程操控已经没问题了，但是机子的显示器是不会显示图形界面的。如果不显示图形界面，可在root权限下输入：
systemctl start gdm.service

即可打开图形界面

牧生

还需要启动配置文件vim /etc/default/grub，在GRUB_CMDLINE_LINUX那行末尾添加rd.driver.blacklist=nouveau nouveau.modeset=0

不做这一步，可能并不能禁用nouveau