1 cuda版本和驱动版本
2 禁用系统自带的显卡驱动:
#新建一个配置文件
sudo vim /etc/modprobe.d/blacklist-nouveau.conf
#写入以下内容
blacklist nouveau
options nouveau modeset=0
#保存并退出
:wq
#重新生成initramfs
sudo dracut --force
#备份当前的镜像
sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
#建立新的镜像
sudo dracut /boot/initramfs-$(uname -r).img $(uname -r)
#重启
sudo reboot
#检查是否关闭
lsmod | grep nouveau
注意:无任何信息输出表示禁用成功;
3 检查驱动
#安装elrepo源
yum -y install https://www.elrepo.org/elrepo-release-7.0-4.el7.elrepo.noarch.rpm
#安装nvidia-detect
yum -y install nvidia-detect
#检测显卡驱动
nvidia-detect -v
4 预安装一些必需的组件
yum install gcc kernel-devel kernel-headers
5 安装显卡驱动
Linux版NVIDIA显卡驱动版本列表
https://download.nvidia.com/XFree86/Linux-x86_64/
5.1 安装对应显卡驱动
可执行权限
chmod +x NVIDIA-Linux-x86_64-xxx.run
#安装
sh ./NVIDIA-Linux-x86_64-xxx.run
5.2 检测显卡是否安装成功:
nvidia-smi
备注:
#安装显卡版本过低错容易出现以下错误
ERROR: Unable to load the 'nvidia-drm' kernel module.
6 安装CUDA
6.1 官网下载对应版本:https://developer.nvidia.com/cuda-toolkit-archive
6.2 复制链接,使用wget下载:
6.3 下载
wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_460.27.04_linux.run
6.4 用sh命令安装CUDA
sh cuda_11.7.0_460.27.04_linux.run
6.4.1 出现一个文档,输入 accept
6.4.2 取消选中Driver ,已经安装了显卡驱动之后,这里会导致冲突报错,一定要取消!
6.4.3 确认无误之后,选择 Install
6.4.4 安装完成
6.4.5 添加环境变量
vim ~/.bashrc
6.4.6 在 .bashrc 文件最后写入并保存:
export PATH=/usr/local/cuda-11.7/bin:$PATH
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.7/lib64
6.4.7 环境变量立即生效
source ~/.bashrc
6.4.8 测试 CUDA 命令,显示如下证明成功!
cat /proc/driver/nvidia/version
nvcc -V
7 CUDNN下载安装
7.1 下载地址:https://developer.nvidia.com/rdp/cudnn-download
7.2 下载后用tar -xvf
解压
#手动下载文件并上传到服务器,然后使用以下命令解压
tar -xvf cudnn-linux-x86_64-8.9.4.25_cuda11-archive.tar.xz
7.3 使用指令进行对应文件的复制即可
sudo cp include/cudnn*.h /usr/local/cuda-11.7/include
sudo cp lib/libcudnn* /usr/local/cuda-11.7/lib64
sudo chmod a+r /usr/local/cuda-11.7/include/cudnn*.h
sudo chmod a+r /usr/local/cuda-11.7/lib64/libcudnn*
7.4 执行以下命令,检查cudnn的安装。
cat /usr/local/cuda-11.7/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
如出现以下输出,则安装成功。
创作不易,求关注,点赞,收藏,谢谢~