本文在安装过程中参考了如下文章,在此向如下博主表示感谢!:
1. https://www.cnblogs.com/timxgb/p/4457265.html
2. https://blog.csdn.net/qq_41493990/article/details/82183035?utm_source=blogxgwz5
3. https://blog.csdn.net/weixin_40298200/article/details/79420758
Nvidia 驱动安装前准备
1 在官网下载相应驱动
网址: https://www.nvidia.cn/Download/Find.aspx?lang=cn
驱动: NVIDIA-Linux-x86_64-396.45.run
cuda 网址: https://developer.nvidia.com/cuda-toolkit-archive
cuda: cuda_9.0.176_384.81_linux.run
cudnn 网址:https://developer.nvidia.com/rdp/cudnn-archive
cudnn: cudnn-9.0-linux-x64-v7.tgz
2 屏蔽默认带有的nouvea
切换到root用户: su root
打开:/lib/modprobe.d/dist-blacklist.conf
将nvidiafb注释掉。
#blacklist nvidiafb然后添加以下语句:
blacklist nouveau
options nouveau modeset=03 重建initramfs image步骤
mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
dracut /boot/initramfs-$(uname -r).img $(uname -r)4 修改运行级别为文本模式
systemctl set-default multi-user.target
5 重新启动, 使用root用户登陆
reboot
6 查看nouveau是否已经禁用
ls mod | grep nouveau
Nvidia 驱动安装
找到驱动 NVIDIA-Linux-x86_64-396.45.run 路径
由于本人电脑有两个显卡,不能安装 opengl,否则重启后图形界面进不去
执行 (root 用户): sh NVIDIA-Linux-x86_64-396.45.run --no-opengl-files
cudnn 9.0 安装
找到 cuda_9.0.176_384.81_linux.run 路径
执行 (root 用户): sh cuda_9.0.176_384.81_linux.run --no-opengl-libs
直接按【Q】键,并输入【accept】回车后
在安装 nvidia 驱动时(Linux-x86_64 384.81),我选择no, 实际上前边已经安装过更高版本驱动了,可以不用安装(我这里也尝试选择yes 安装,但没有安装成功).
cudnn 安装
找到 cudnn-9.0-linux-x64-v7.tgz 路径
执行: tar -xzvf cudnn-9.0-linux-x64-v7.tgz
执行 (root 用户):cp cuda/include/cudnn.h /usr/local/cuda/include
执行 (root 用户):cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
环境变量
在~/.bashrc 的最后添加
export PATH=/usr/local/cuda/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda
CUDNN连接建立
cd /usr/local/cuda/lib64
sudo rm -rf libcudnn.so libcudnn.so.7 #删除原有版本号,版本号在cudnn/lib64中查询
sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7 #生成软连接,注意自己下载的版本号
sudo ln -s libcudnn.so.7 libcudnn.so
sudo ldconfig #立即生效