nvidia gpu是目前人工智能常用的硬件,使用该硬件,一般安装程序适配的cuda即可(cuda中含有GPU的驱动),如果要单独安装驱动,也可以先安装驱动,再安装cuda,cuda安装过程中,把驱动安装步骤取消掉即可。
1、首先到官网上找到自己GPU硬件对应的cuda版本:
CUDA Toolkit Archive | NVIDIA Developer
2、安装前设置
# 1.关闭图形界面
#systemctl set-default multi-user.target
#systemctl stop graphical.target
# 2.禁用系统自带的驱动nouveau
# shell执行:
#lsmod | grep nouveau
#如果上述脚本执行结果为空,则可以跳过本步骤。
#vim /etc/modprobe.d/blacklist.conf
添加:
blacklist nouveau
options nouveau modeset=0
#vim /usr/lib/modprobe.d/dist-blacklist.conf
添加:
blacklist nouveau
options nouveau modeset=0
#rmmod nouveau
# 查看nouveau,保障nouveau已不存在
#lsmod | grep nouveau
3、执行安装
#chmod 755 cuda_11.0.2_450.51.05_linux.run
#./cuda_11.0.2_450.51.05_linux.run
进入图形安装界面,按提示进行操作。如有报错,根据报错提示修改。