Ubuntu处理nvidia显卡驱动相关内容学习笔记
1.Ubuntu驱动安装
1.1 查找推荐版本
>> sudo ubuntu-drivers devices
== /sys/devices/pci0000:64/0000:64:00.0/0000:65:00.0 ==
modalias : pci:v000010DEd00002484sv00001028sd00003907bc03sc00i00
vendor : NVIDIA Corporation
model : GA104 [GeForce RTX 3070]
driver : nvidia-driver-470 - distro non-free
driver : nvidia-driver-570-server-open - distro non-free
driver : nvidia-driver-570-open - distro non-free
driver : nvidia-driver-570 - distro non-free recommended
driver : nvidia-driver-535-server - distro non-free
driver : nvidia-driver-545 - distro non-free
driver : nvidia-driver-535 - distro non-free
driver : nvidia-driver-470-server - distro non-free
driver : nvidia-driver-570-server - distro non-free
driver : nvidia-driver-550-open - third-party non-free
driver : nvidia-driver-545-open - distro non-free
driver : nvidia-driver-535-server-open - distro non-free
driver : nvidia-driver-535-open - distro non-free
driver : nvidia-driver-550 - third-party non-free
driver : xserver-xorg-video-nouveau - distro free builtin
1.2 安装推荐版本
看一下recommanded那一项，然后安装对应版本
>> sudo apt-get install nvidia-driver-570
2.nvidia的cuda安装
- 不同版本安装命令可以在官网选择对应选项生成：https://developer.nvidia.com/cuda-toolkit-archive
>> wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-ubuntu2204.pin
>> sudo mv cuda-ubuntu2204.pin /etc/apt/preferences.d/cuda-repository-pin-600
>> wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb
>> sudo dpkg -i cuda-repo-ubuntu2204-12-4-local_12.4.0-550.54.14-1_amd64.deb
>> sudo cp /var/cuda-repo-ubuntu2204-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
>> sudo apt-get update
>> sudo apt-get -y install cuda-toolkit-12-4
3.nvidia的cudnn安装
- 不同版本安装命令可以在官网选择第应选项生成：https://developer.nvidia.com/cudnn-archive
>> wget https://developer.download.nvidia.com/compute/cudnn/9.6.0/local_installers/cudnn-local-repo-ubuntu2204-9.6.0_1.0-1_amd64.deb
>> sudo dpkg -i cudnn-local-repo-ubuntu2204-9.6.0_1.0-1_amd64.deb
>> sudo cp /var/cudnn-local-repo-ubuntu2204-9.6.0/cudnn-*-keyring.gpg /usr/share/keyrings/
>> sudo apt-get update
>> sudo apt-get -y install cudnn
查看相关文件路径：
>> sudo find / -name "cudnn*.h" # 查找头文件
>> sudo find / -name "libcudnn*" # 查找库文件
可以看到：
- 头文件：/usr/include
- 库文件：/usr/lib/x86_64-linux-gnu（64 位系统）
CMake中使用时：
set(cuDNN_LIBRARY_DIR "/usr/lib/x86_64-linux-gnu" CACHE PATH "cuDNN library directory")
set(cuDNN_INCLUDE_DIR "/usr/include" CACHE PATH "cuDNN include directory")
find_package(cuDNN REQUIRED)
4.nvidia的tensorrt安装
- 不同版本下载地址：https://developer.nvidia.com/tensorrt/download
>> wget https://developer.nvidia.com/downloads/compute/machine-learning/tensorrt/10.6.0/local_repo/nv-tensorrt-local-repo-ubuntu2204-10.6.0-cuda-12.6_1.0-1_amd64.deb
>> sudo dpkg -i nv-tensorrt-local-repo-ubuntu2204-10.6.0-cuda-12.6_1.0-1_amd64.deb
# 上面安装完，会生成下面的提示
>> sudo cp /var/nv-tensorrt-local-repo-ubuntu2204-10.6.0-cuda-12.6/nv-tensorrt-local-2C8302C3-keyring.gpg /usr/share/keyrings/
# 安装tensorrt
>> sudo apt-get install tensorrt
安装的相关路径为：
- (1) TensorRT推理工具：/usr/src/tensorrt/bin/trtexec
- (2) TensorRT头文件路径：/usr/include/x86_64-linux-gnu/
- (3) TensorRT库文件路径：/usr/lib/x86_64-linux-gnu/
相关查找指令：
sudo find / -name "libnvinfer.so*" # 搜索 TensorRT 库文件
sudo find / -name "NvInfer*.h" # 搜索 TensorRT 头文件