CentOS7安装NVIDIA显卡驱动+CUDA+CuDNN

1 cuda版本和驱动版本

2 禁用系统自带的显卡驱动：

#新建一个配置文件
sudo vim /etc/modprobe.d/blacklist-nouveau.conf

#写入以下内容
blacklist nouveau
options nouveau modeset=0

#保存并退出
:wq

#重新生成initramfs
sudo dracut --force

#备份当前的镜像
sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak

#建立新的镜像
sudo dracut /boot/initramfs-$(uname -r).img $(uname -r)

#重启
sudo reboot

#检查是否关闭
lsmod | grep nouveau

注意：无任何信息输出表示禁用成功；

3 检查驱动

#安装elrepo源
yum -y install https://www.elrepo.org/elrepo-release-7.0-4.el7.elrepo.noarch.rpm

#安装nvidia-detect
yum -y install nvidia-detect

#检测显卡驱动
nvidia-detect -v

4 预安装一些必需的组件

yum install gcc kernel-devel kernel-headers

5 安装显卡驱动

Linux版NVIDIA显卡驱动版本列表
https://download.nvidia.com/XFree86/Linux-x86_64/

5.1 安装对应显卡驱动

可执行权限
chmod +x NVIDIA-Linux-x86_64-xxx.run

#安装
sh ./NVIDIA-Linux-x86_64-xxx.run

5.2 检测显卡是否安装成功：

nvidia-smi

备注：
#安装显卡版本过低错容易出现以下错误

ERROR: Unable to load the 'nvidia-drm' kernel module.  

6 安装CUDA

6.1 官网下载对应版本：https://developer.nvidia.com/cuda-toolkit-archive

6.2 复制链接，使用wget下载：

6.3 下载

wget https://developer.download.nvidia.com/compute/cuda/11.2.0/local_installers/cuda_11.2.0_460.27.04_linux.run

6.4 用sh命令安装CUDA

sh cuda_11.2.0_460.27.04_linux.run

6.4.1 出现一个文档，输入 accept

6.4.2 取消选中Driver ，已经安装了显卡驱动之后，这里会导致冲突报错，一定要取消！

6.4.3 确认无误之后，选择 Install

6.4.4 安装完成

6.4.5 添加环境变量

vim ~/.bashrc

6.4.6 在 .bashrc 文件最后写入并保存：

export PATH=/usr/local/cuda-11.2/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.2/lib64

6.4.7 环境变量立即生效

source ~/.bashrc

6.4.8 测试 CUDA 命令，显示如下证明成功！

cat /proc/driver/nvidia/version
nvcc -V

7 CUDNN下载安装

7.1 下载地址：https://developer.nvidia.com/rdp/cudnn-download
7.2 下载后用tar -xvf解压

7.3 使用指令进行对应文件的复制即可

sudo cp include/cudnn*.h /usr/local/cuda-11.2/include/ 
sudo cp lib/libcudnn* /usr/local/cuda-11.2/lib64/ 
sudo chmod a+r /usr/local/cuda-11.2/include/cudnn.h 
sudo chmod a+r /usr/local/cuda-11.2/lib64/libcudnn*

7.4 执行以下命令，检查cudnn的安装。

cat /usr/local/cuda-11.2/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

如出现以下输出，则安装成功。

创作不易，求关注，点赞，收藏，谢谢~