Linux安装NVIDIA显卡驱动和CUDA Toolkit在Linux服务器上安装NVIDIA显卡驱动和CUDA To

本文目标

在Linux服务器上安装NVIDIA显卡驱动和CUDA Toolkit

一、安装显卡驱动

1. 确定显卡型号(可选)

如果知道显卡型号，可以跳过这一步

lspci | grep -i nvidia

执行上面的命令，应该输出如下图所示，2b85就是型号

然后到这里输入型号进行查询：链接

2. 下载驱动

到官网下载驱动：链接

3. 驱动安装

准备工作

#安装前置依赖
sudo apt-get update
sudo apt-get install g++
sudo apt-get install gcc
sudo apt-get install make

#卸载旧驱动
sudo apt-get remove --purge nvidia* 

#禁用默认驱动
sudo vim /etc/modprobe.d/blacklist.conf
#在文件末尾增加下面两行
blacklist nouveau
options nouveau modeset=0

#更新文件
sudo update-initramfs -u

#重启服务器
reboot

开始安装

#创建文件件
mkdir -p /usr/nvidia

#把下载的驱动上传到这个文件夹（过程略）

#授权 xxx是文件名
sudo chmod 777 xxx.run

#安装
sudo ./NVIDIA-Linux-x86_64-550.78.run

4. 验证

nvidia-smi

安装成功

二、安装CUDA Toolkit

1.安装

wget https://developer.download.nvidia.com/compute/cuda/12.8.0/local_installers/cuda_12.8.0_570.86.10_linux.runsudo
sh cuda_12.8.0_570.86.10_linux.run

如果需要其它版本，可以到官网找其他版本的安装命令：链接

2.配置

sudo vim ~/.bashrc

#在文件末尾添加（主要路径是否正确，如果安装其它版本，可能不是这个路径）
export PATH="/usr/local/cuda-12.8/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-12.8/lib64:$LD_LIBRARY_PATH"

#配置生效
source ~/.bashrc

3.验证

nvcc -V