NVIDIA显卡驱动相关

发布时间 2023-12-26 11:32:02作者: 摆科全书

cuda driver install

准备阶段

首先确认 cuda xserver 版本,保证和自己电脑显卡、cuda toolkit 版本适配,相关信息:
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html

然后下载 driver 文件:
https://www.nvidia.cn/geforce/drivers/

卸载之前版本 nvidia driver :

sudo apt-get remove --purge nvidia*

一般来说,默认 driver 是 nouveau (对于ubuntu), 所以它需要被禁用 (注意修改版本对应路径):

cd /lib/modules/x.x.x-y-generic/kernel/drivers/gpu/drm/nouveau
mv nouveau.ko nouveau.ko.org
update-initramfs -u

或者

sudo vim /etc/modprobe.d/blacklist.conf

然后把这两行加在文件最后:

	blacklist nouveau
	options nouveau modeset=0
// 保存后执行
update-initramfs -u

重新生成 kernel initramfs 并且重启,最后检查:

lsmod | grep nouveau

没显示信息的话就拿捏了

安装

开始安装:

sudo chmod a+x NVIDIA-Linux-x86_64-xxx.xx.run
sudo ./NVIDIA-Linux-x86_64-xxx.xx.run -no-opengl-files

最后检查显卡能不能被正常识别,在设置里面看,或者用命令:

nvidia-smi

cuda toolkit install

到官网,用对应版本的命令:
https://developer.nvidia.com/cuda-toolkit-archive
在下载之前先复制 key ,不然他会报错,这里我用官网步骤就是这样,按报错信息稍微变换一下命令执行顺序就行。

然后更新配置文件:

sudo vim ~/.bashrc

把这两行加到最后:

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

最后,再次更新配置文件并且检查其是否正常安装:

source ~/.bashrc
nvcc --version

如果找不到 nvcc 的话, 在 bashrc 文件加上这条命令,再更新一次:

export PATH=$PATH:/usr/local/cuda/bin

确保这个目录是可以找到 nvcc 可执行文件的。

cudnn install

到官网下载:
https://developer.nvidia.com/rdp/cudnn-download

按照 installation guide 进行, 如果在用 mnistCUDNN 测试时有这种报错:

fatal error: FreeImage.h: No such file or directory

那就安装依赖包:

sudo apt-get install libfreeimage3 libfreeimage-dev

之后理论上就都OK了