当前位置:   article > 正文

CentOS7安装NVIDIA显卡驱动+CUDA+CuDNN_centos7安装cudnn

centos7安装cudnn

1 cuda版本和驱动版本

在这里插入图片描述

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

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

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

#保存并退出
:wq
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
  • 9
#重新生成initramfs
sudo dracut --force
  • 1
  • 2
#备份当前的镜像
sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak

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

#重启
sudo reboot
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8
#检查是否关闭
lsmod | grep nouveau
  • 1
  • 2

注意:无任何信息输出表示禁用成功;

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
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8

4 预安装一些必需的组件

yum install gcc kernel-devel kernel-headers
  • 1

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
  • 1
  • 2
  • 3
  • 4
  • 5

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

nvidia-smi
  • 1

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

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

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
  • 1

6.4 用sh命令安装CUDA

sh cuda_11.2.0_460.27.04_linux.run
  • 1

6.4.1 出现一个文档,输入 accept
在这里插入图片描述

6.4.2 取消选中Driver ,已经安装了显卡驱动之后,这里会导致冲突报错,一定要取消!
在这里插入图片描述
6.4.3 确认无误之后,选择 Install
在这里插入图片描述
6.4.4 安装完成
在这里插入图片描述
6.4.5 添加环境变量

vim ~/.bashrc
  • 1

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
  • 1
  • 2

6.4.7 环境变量立即生效

source ~/.bashrc
  • 1

6.4.8 测试 CUDA 命令,显示如下证明成功!

cat /proc/driver/nvidia/version
nvcc -V
  • 1
  • 2

7 CUDNN下载安装

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

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

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.2/include
sudo cp cuda/lib64/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*
  • 1
  • 2
  • 3
  • 4

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

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

如出现以下输出,则安装成功。
在这里插入图片描述


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

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/菜鸟追梦旅行/article/detail/128604
推荐阅读
相关标签
  

闽ICP备14008679号