1. 安装Nvidia 显卡驱动
步骤一: 安装更新软件列表和依赖项
在安装Nvidia显卡驱动之前,需要更新软件列表和必要的依赖项。
sudo apt-get update # 更新软件列表
sudo apt-get install g++ # 下载g++编译器
sudo apt-get install gcc # 下载gcc编译器
sudo apt-get install make # 下载GNU Make编译器
sudo apt-get install initramfs-tools # 下载安装initramfs-tools
步骤二:禁用nouveau通用驱动
修改blacklist.conf文件
sudo vim /etc/modprobe.d/blacklist.conf
在blacklist.conf文件,最下方添加命令
blacklist nouveau
options nouveau modeset=0
执行以下命令使禁用生效并且重启:
sudo update-initramfs -u
sudo reboot
重启后可以验证是否生效:
lsmod | grep nouveau
若没有输出,则禁用生效。
步骤三: 查看GPU型号,并下载对应驱动
1.打开终端输入以下命令,获取显卡驱动信息
ubuntu-drivers devices
2.选择推荐的驱动版本进行安装
注意:显卡驱动版本和cuda版本不一样,显卡版本按推荐的配置就好。
sudo apt install nvidia-driver-535
sudo reboot #安装完成后重启
nvidia-smi #如果出现GPU列表,则驱动安装成功
2. 安装CUDA
为什么要安装11.3版本,因为11.3版本适配的pytorch版本是最多的,很多像YOLO模型,需要专门的pytorch版本,否则全是报错
在官网下载cuda,版本根据自己的选择来,依然强烈推荐11.3
[CUDA官网](https://developer.nvidia.com/cuda-toolkit-archi
按图中进行选择
输入对应的命令
wget https://developer.download.nvidia.com/compute/cuda/11.3.0/local_installer/cuda_11.3.0_465.19.01_linux.run
sudo sh cuda_11.3.0_465.19.01_linux.run
选择continue
输入accept
这里有一点需要注意一下,因为之前已经安装过显卡驱动了,所以这里需要取消Driver(光标移动到Driver,然后按enter即可取消,不取消会导致安装终止),然后移动光标到Install,按enter执行安装。
接下来进行环境配置
nano ~/.bashrc #打开bashrc脚本文件
export PATH=/usr/local/cuda-11.3/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.3/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
source ~/.bashrc #脚本文件生效
nvcc -V
安装成功
2. 安装CUDNN
- 下载CUDA版本对应的cuDNN版本 cudnn官网
可以选择11.3或者11.x
cudnn安装包的copy
sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda/include
sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn*.h
验证
cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
显示对应的cudnn 版本是8.9.2,安装成功