硬件设施:

  • centos7

在这里插入图片描述NVidia显卡驱动:470.57
CUDA: 11.4.2
cuDNN: 8.2.4

1. 禁用默认的驱动nouveau

检查 nouveau driver 有没有被加载
在这里插入图片描述

1) 在 /usr/lib/modprobe.d/dist-blacklist.conf 中添加两行内容:

blacklist nouveau
options nouveau modeset=0

2)接着给当前镜像做备份

mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak

3)建立新的镜像

dracut /boot/initramfs-$(uname -r).img $(uname -r)

4)重新启动

5) 检查nouveau driver确保没有被加载

lsmod | grep nouveau

在这里插入图片描述
禁用成功!!

2. 系统要求

首先要保证操作系统符合CUDA 11.2的要求。
在这里插入图片描述

在centos7下安装,内核和centos的版本分别需要是3.10和7.7。使用下面命令检查。

uname -r                             
cat /etc/centos-release

如果版本不满足,需要执行以下命令,更新操作系统。

sudo yum clean all
sudo yum update

更新的时间可能会比较长,请耐性等待,一般都会顺利成功。万一发生错误,到网上找找方案,很快就能解决。

3. 软件更新

1)验证系统内核版本和安装开发包。

uname -r
sudo yum install -y gcc gcc-c++ kernel-devel-$(uname -r) kernel-headers-$(uname -r)

2)验证gcc的版本。

gcc --version

由于CUDA 11.3要求GCC的版本是6以上,下面是安装GCC7的脚本。

sudo yum install centos-release-scl
sudo yum install devtoolset-7

# launch a new shell instance using the Software Collection scl tool:
scl enable devtoolset-7 bash
gcc --version

如果重新登录,gcc版本还是原先的版本,所以在CUDA安装前,需要执行scl enable devtoolset-7 bash。

4. 安装驱动

驱动下载:

nvidia驱动

所有驱动下载地址:

https://download.nvidia.com/XFree86/Linux-x86_64/

wget https://download.nvidia.com/XFree86/Linux-x86_64/470.74/NVIDIA-Linux-x86_64-470.74.run

相应的软件下载:
链接: https://pan.baidu.com/s/1NW056ateq4IN8BPrisz6rg 密码: mcgg

在这里插入图片描述

在这里插入图片描述

nvidia-smi

在这里插入图片描述

5. 安装cuda

在这里插入图片描述运行安装文件

chmod 755 cuda_11.4.2_470.57.02_linux.run
 ./cuda_11.4.2_470.57.02_linux.run 

在这里插入图片描述不要选择驱动
在这里插入图片描述
安装成功

在这里插入图片描述

6. 安装cudnn

解压命令为:

tar zxvf cudnn-11.4-linux-x64-v8.2.4.15.tgz -C .

在这里插入图片描述

并进入到相应目录,运行以下命令:

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

在这里插入图片描述

nvcc -V

在这里插入图片描述

以配置cuDNN环境
以上则配置完成!

参考

https://eipi10.cn/deep-learning/2019/11/28/centos_cuda_cudnn/

Logo

华为开发者空间,是为全球开发者打造的专属开发空间,汇聚了华为优质开发资源及工具,致力于让每一位开发者拥有一台云主机,基于华为根生态开发、创新。

更多推荐