centos7.6 安装 cuda驱动

1、首先确认 nouveau 模块有没有启用，如果启用的话禁掉。

2、如果是高版本的系统，则需要高版本的nvidia cuda以及cudnn

nvidia驱动官网
https://www.nvidia.com/Download/index.aspx?spm=a2c63.p38356.879954.7.2c524526OCooRk&lang=cn

选择驱动要根据实际的GPU卡型号选，可以参考阿里云上的文章：
https://www.alibabacloud.com/help/zh/doc-detail/108502.htm

cuda驱动下载网址
https://developer.nvidia.com/cuda-downloads

比如下载10.0 10.1 相对高版本的驱动程序

cudnn也选择较高版本，例如10.0
cudnn下载地址
https://developer.nvidia.com/cudnn

3、安装驱动
3.1、安装nvidia驱动

sh NVIDIA-Linux-x86_64-410.129-diagnostic

 

验证

nvidia-smi

 

3.2、安装cuda驱动
yum -y install gcc kernel-devel "kernel-devel-uname-r == $(uname -r)" dkms

sh cuda_10.0.130_410_48_linux.run

Do you accept the previously read EULA?
accept/decline/quit: accept

Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 396.37?
(y)