和GPU有关的组件有NVIDIA,cuda,cudnn,gpu
其中gpu是硬件,cuda是gpu的并行计算框架。
cudnn是cuda的运算加速库。
NVIDIA是具体显卡的驱动。
因此安装显卡驱动时,可以按照如下方式安装
- 先安装cuda。
其中安装cuda时,可以选择安装cuda里的NVIDIA驱动,也可以自己装一个NVIDIA驱动。
选项install nvidia accelerated Graphics Driver,输入n,不装cuda里默认的nvidia的驱动。
自己装NVIDIA驱动时,可以按照不同cuda版本的驱动。(不确定是否可以只安装NVIDIA,而不安装cuda) - 安装NVIDIA。
显卡驱动安装
1、lspci | grep -i vga : 查看显卡号
2、登录 https://pci-ids.ucw.cz/read/PC/10de 查询显卡号对应显卡型号
3、登录 https://www.nvidia.cn/Download/index.aspx?lang=cn 查询显卡型号对应驱动并下载
4、uname -r : 查询内核版本号
5、yum install kernel-devel-内核版本号 : 安装kernel-devel包
6、sh NVIDIA*.run --kernel-source-path=/usr/src/kernels/内核版本号/