nvidia驱动安装配置

安装配置nvidia驱动及cuda环境，cudnn安装配置。

安装驱动

环境

ubuntu 20.04
启动项 secure boot 关闭

禁用nouveau

编辑文件blacklist.conf vim /etc/modprobe.d/blacklist.conf 在文件最后部分插入以下两行内容 blacklist nouveau options nouveau modeset=0

更新系统

sudo update-initramfs -u

重启系统

sudo reboot

环境部署

在英伟达的官网上查找你自己电脑的显卡型号然后下载相应的驱动。网址：https://www.nvidia.cn/Download/index.aspx?lang=cn

可执行文件赋权

sudo chmod  a+x NVIDIA-Linux-x86_64-396.18.run

安装依赖

sudo apt install gcc make -y

运行安装脚本，根据安装提示安装

sudo ./NVIDIA-Linux-x86_64-396.18.run

安装cuda环境

同样在官网下载选择合适的版本 https://developer.nvidia.com/cuda-12-4-1-download-archive?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=20.04&target_type=runfile_local

根据官方给的命令行下载，在安装过程中选择Continue,输入accept,使用空格去掉Driver,最后install安装。安装完成后编辑用户环境变量，在~/.bashrc文件中添加以下内容

export PATH=/usr/local/cuda-12.4/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-12.4/lib64:$LD_LIBRARY_PATH

nvcc -V 验证cuda是否安装成功

安装cudnn

1.选择合适的下载版本 https://developer.nvidia.cn/rdp/cudnn-archive

2.下载cudnn库文件，解压到指定目录

cp lib/* /usr/local/cuda-12.4/lib64/
cp include/* /usr/local/cuda-12.4/include/

3.添加执行权限

chmod a+r /usr/local/cuda-12.4/include/cudnn.h /usr/local/cuda-12.4/lib64/libcudnn*

安装 NVIDIA Docker

1.添加 NVIDIA Docker 仓库

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-docker/gpgkey | sudo apt-key add -
curl -s -L https://nvidia.github.io/nvidia-docker/$distribution/nvidia-docker.list | sudo tee /etc/apt/sources.list.d/nvidia-docker.list

2.安装

sudo apt-get update
sudo apt-get install -y nvidia-docker2
sudo systemctl restart docker

文档信息

本文作者：onefeng
本文链接：https://me.onefeng.xyz/2023/08/05/nvidia%E9%A9%B1%E5%8A%A8%E5%AE%89%E8%A3%85%E9%85%8D%E7%BD%AE/
版权声明：自由转载-非商用-非衍生-保持署名（创意共享3.0许可证）

ONEFENG