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)