【技术】TensorRT 10.7 安装指南(Ubuntu22.04)

原文链接:https://mengwoods.github.io/post/tech/008-tensorrt-installation/

本文安装的版本如下:

Ubuntu 22.04

Nvidia Driver 538.78

CUDA 12.2

cuDNN 8.9.7

TensorRT 10.7

安装前的准备(可选)

在安装新版本之前,先卸载已有的 NVIDIA 驱动、CUDA、cuDNN 和 TensorRT,以避免冲突。

bash 复制代码
sudo apt remove --purge '^nvidia-.*' '^cuda.*' '^libcuda.*' '^libnvidia.*' '^tensorrt.*' '^cudnn.*'
sudo apt autoremove -y
sudo apt clean
# Remove CUDA Toolkit and Related Files
sudo rm -rf /usr/local/cuda*
sudo rm -rf /usr/lib/cuda*
# Remove NVIDIA Repository (If Installed)
sudo rm -f /etc/apt/sources.list.d/cuda*
sudo rm -f /etc/apt/sources.list.d/nvidia-*
# update your package lis
sudo apt update

验证卸载:

bash 复制代码
dpkg -l | grep -E 'cuda|nvidia|tensorrt|cudnn'

重新启动系统:

bash 复制代码
sudo reboot

重新启动后,检查是否卸载干净:

bash 复制代码
# Check Nvidia driver
nvidia-smi
# check CUDA
which nvcc

安装 NVIDIA 驱动

https://www.nvidia.com/en-us/drivers/

bash 复制代码
# install the latest supported driver
sudo ubuntu-drivers autoinstall
# install the specific version
sudo apt install nvidia-driver-xxx

安装 CUDA v12.2

https://developer.nvidia.com/cuda-12-2-0-download-archive

前往 CUDA 下载页面 获取安装说明,或直接使用以下命令(适用于 Ubuntu 22.04 x86_64)

bash 复制代码
wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
sudo dpkg -i cuda-keyring_1.1-1_all.deb
sudo apt-get update
sudo apt-get -y install cuda
sudo apt-get install nvidia-cuda-toolkit

配置环境变量

bash 复制代码
# Update cuda path to .bashrc
echo 'export PATH=/usr/local/cuda-12.8/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-12.8/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc
# 验证
nvcc --version

安装 cuDNN v8.9.7

https://developer.nvidia.com/rdp/cudnn-archive

前往 cuDNN 归档页面 下载对应 .deb 文件。

bash 复制代码
# 安装 cuDNN 相关的本地仓库
sudo dpkg -i cudnn-local-repo-ubuntu2004-8.9.7.29_1.0-1_amd64.deb
# 复制密钥(请根据你的文件夹路径调整)
sudo cp /var/cudnn-local-repo-ubuntu2004-8.9.7.29/cudnn-*-keyring.gpg /usr/share/keyrings/
# update and install
sudo apt-get update
sudo apt-get install libcudnn8
# verify installation
dpkg -l | grep cudnn

安装 TensorRT 10.7

https://developer.nvidia.com/tensorrt-getting-started

前往 TensorRT 官方页面 下载 .deb 文件。

bash 复制代码
# 安装 TensorRT 本地仓库(请确认文件名是否一致)
sudo dpkg -i nv-tensorrt-local-repo-ubuntu2204-10.7.0-cuda-12.6_1.0-1_amd64.deb
# 复制密钥(请确认文件夹路径是否正确)
sudo cp /var/nv-tensorrt-local-repo-ubuntu2204-10.7.0-cuda-12.6/nv-tensorrt-local-F234AD55-keyring.gpg /usr/share/keyrings/
# update package list
sudo apt-get update
# 安装 TensorRT
sudo apt-get install tensorrt
# 添加环境变量
echo 'export PATH=$PATH:/usr/src/tensorrt/bin' >> ~/.bashrc && source ~/.bashrc
source ~/.bashrc
# 验证安装
dpkg -l | grep nvinfer
sudo apt show tensorrt
trtexec --version
相关推荐
go54631584655 分钟前
基于深度学习的食管癌右喉返神经旁淋巴结预测系统研究
图像处理·人工智能·深度学习·神经网络·算法
Blossom.1187 分钟前
基于深度学习的图像分类:使用Capsule Networks实现高效分类
人工智能·python·深度学习·神经网络·机器学习·分类·数据挖掘
宇称不守恒4.09 分钟前
2025暑期—05神经网络-卷积神经网络
深度学习·神经网络·cnn
格林威1 小时前
Baumer工业相机堡盟工业相机如何通过YoloV8深度学习模型实现沙滩小人检测识别(C#代码UI界面版)
人工智能·深度学习·数码相机·yolo·计算机视觉
巫婆理发2222 小时前
神经网络(多层感知机)(第二课第二周)
人工智能·深度学习·神经网络
xw33734095642 小时前
彩色转灰度的核心逻辑:三种经典方法及原理对比
人工智能·python·深度学习·opencv·计算机视觉
贝塔西塔3 小时前
PytorchLightning最佳实践基础篇
pytorch·深度学习·lightning·编程框架
ReinaXue4 小时前
大模型【进阶】(五):低秩适配矩阵LORA的深度认识
人工智能·深度学习·神经网络·语言模型·自然语言处理·transformer
DisonTangor4 小时前
商汤InternLM发布最先进的开源多模态推理模型——Intern-S1
人工智能·深度学习·开源·aigc
都叫我大帅哥4 小时前
深度学习的"Hello World":多层感知机全解指南
python·深度学习