CentOS安装4090显卡驱动和CUDA工具包
更新软件安装源
yum -y update --exclude=kernel* --exclude=centos-release* --exclude=initscripts*
yum install -y kernel-devel-`uname -r` gcc gcc-c++
官网地址 https://developer.nvidia.com/cuda-toolkit-archive
首先下载CUDA工具包包含了4090驱动
wget https://developer.download.nvidia.cn/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run
当然你也可以单独下载显卡驱动安装,特殊用途的还是需要安装显卡驱动
https://www.nvidia.cn/Download/index.aspx?lang=cn
如果系统没有wget是无法下载可以执行
yum -y install wget
wget https://cn.download.nvidia.cn/XFree86/Linux-x86_64/535.104.05/NVIDIA-Linux-x86_64-535.104.05.run
由于很多软件需要配套的版本,所以要下载一样的版本最好
执行以下命令,查看是否安装Nouveau驱动
lsmod | grep nouveau
如果回显信息中包含Nouveau驱动信息,说明Nouveau驱动已安装,则需要禁用Nouveau驱动 执行如下命令编辑blacklist.conf文件。 如果没有“/etc/modprobe.d/blacklist.conf”文件,请新建一个。
vi /etc/modprobe.d/blacklist.conf
添加如下语句添加至文件结尾
blacklist nouveau
options nouveau modeset=0
关闭文件后执行以下命令,备份并新建一个initramfs。
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
dracut -v /boot/initramfs-$(uname -r).img $(uname -r)
然后重启机器 reboot 再次安装显卡驱动。
sh cuda_12.2.2_535.104.05_linux.run
为了简单,我直接安装套件包会与驱动一同安装,简单省事。
同意安装协议 输入 accept 然后回车下一步
用方向键移动到安装,直接回车等待一会即可。
然后指定CUDA环境,编辑文件
vim /etc/bashrc
在最后面添加
export CUDA_HOME=/usr/local/cuda-12.2
export LD_LIBRARY_PATH=${CUDA_HOME}/lib64:${LD_LIBRARY_PATH}
export PATH=${CUDA_HOME}/bin:${PATH}
注意你安装的版本号,然后重新加载CUDA环境变量
source /etc/bashrc
至此结束了,检测
nvcc -V
出现相关信息
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2023 NVIDIA Corporation Built on Tue_Aug_15_22:02:13_PDT_2023 Cuda compilation tools, release 12.2, V12.2.140 Build cuda_12.2.r12.2/compiler.33191640_0
在检测下显卡驱动 nvidia-smi
会显示相关显卡驱动信息,到此结束