关于我们

质量为本、客户为根、勇于拼搏、务实创新

新闻公告

< 返回新闻公共列表

CentOS安装4090显卡驱动和CUDA工具包

发布时间:2023-12-15 12:03:22

更新软件安装源

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

<span style="color: rgb(18, 18, 18);font-family: -apple-system, BlinkMacSystemFont," segoe="" ui",="" roboto,="" oxygen,="" ubuntu,="" cantarell,="" "fira="" sans",="" "droid="" "helvetica="" neue",="" sans-serif;="" font-size:="" medium;="" white-space:="" pre;"="">在检测下显卡驱动

nvidia-smi

会显示相关显卡驱动信息,到此结束 



/template/Home/Zkeys/PC/Static