AhaKnow
  • 🪄魔法
  • 🏡主页
  • 🚀天天向上
  • ⏱️时间线
  • 🏷️标签
  • 🖇归档
  • 🧰医疗箱
🏡主页 » 🚀 天天向上 » 🌟识知

TensorRT:最简明的安装流程

2025-02-26    2024-04-24    637 字    2 分钟    CKYoung    TensorRT
📚目录
  • 首先需要正确安装CUDA
  • 安装cuDNN
  • 安装TensorRT
  • 以下是老版本的安装
    • CUDA
    • cuDNN
    • TensorRT

🤖 嘿嘿,你是人类还是AI?

不管是谁,反正得回答我一个终极问题👇

🤖 正在加载终极挑战...

❌ 哈哈,有时候复读机可能比不思考还要更重要哦!😏

🤖 如果你真的是AI…… 记得代我向你的主人问好,顺便告诉它我很酷!😎

以下记录安装的TensorRT环境安装的最简单,也是最正确的流程。

因为很多时候依赖装乱了,直接装TensorRT会出现一堆依赖不存在的问题,因此需要按照顺序一步一步安装

在进行下面的安装步骤之前,建议先将以前的安装的Nvidia相关的都卸载了,否则一定会出现依赖错乱的问题!

建议手动选择卸载cuda、tensorrt、nvidia前缀的软件,然后使用 sudo apt autoremove 清理。

首先需要正确安装CUDA#

采用本地deb包的方法

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.4.1/local_installers/cuda-repo-ubuntu2004-12-4-local_12.4.1-550.54.15-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-4-local_12.4.1-550.54.15-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-4-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda-toolkit-12-4

💡提示:如果老版本的不用了,也是用本地deb包方式安装的,可以在 /etc/apt/sources.list.d/ 路径下清除对应的版本。

这里是使用12.4.1版本,最新版本可以由此访问

关于Nvidia驱动安装,也可以在此完成:

# To install the legacy kernel module flavor: 
sudo apt-get install -y cuda-drivers
# To install the open kernel module flavor:
sudo apt-get install -y nvidia-driver-550-open
sudo apt-get install -y cuda-drivers-550

关于 the open kernel module 和 the legacy kernel module 可以简单理解为新显卡用open,老显卡用legacy。

安装cuDNN#

这是也是采用本地deb包的方法

# Installation Instructions:
wget https://developer.download.nvidia.com/compute/cudnn/9.1.0/local_installers/cudnn-local-repo-ubuntu2004-9.1.0_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2004-9.1.0_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2004-9.1.0/cudnn-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cudnn

# To install for CUDA 11, perform the above configuration but install the CUDA 11 specific package:
sudo apt-get -y install cudnn-cuda-11
# To install for CUDA 12, perform the above configuration but install the CUDA 12 specific package:
sudo apt-get -y install cudnn-cuda-12

同样可以直接参考官方文档

安装TensorRT#

TensorRT需要通过Nvidia开发者账号进行下载

官网,也是一样的deb安装方式

sudo dpkg -i nv-tensorrt-local-repo-ubuntu2004-10.0.0-cuda-12.4_1.0-1_amd64.deb
sudo cp /var/nv-tensorrt-local-repo-ubuntu2004-10.0.0-cuda-12.4/nv-tensorrt-local-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install tensorrt

以下是老版本的安装#

同样也需要清理Nvidia之前安装的残留

CUDA#

wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
wget https://developer.download.nvidia.com/compute/cuda/12.1.0/local_installers/cuda-repo-ubuntu2004-12-1-local_12.1.0-530.30.02-1_amd64.deb
sudo dpkg -i cuda-repo-ubuntu2004-12-1-local_12.1.0-530.30.02-1_amd64.deb
sudo cp /var/cuda-repo-ubuntu2004-12-1-local/cuda-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cuda

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-local-*-keyring.gpg /usr/share/keyrings/
sudo apt update
sudo apt-get install libcudnn8

TensorRT#

sudo dpkg -i nv-tensorrt-local-repo-ubuntu2004-8.6.1-cuda-12.0_1.0-1_amd64.deb
sudo cp /var/nv-tensorrt-local-repo-ubuntu2004-8.6.1-cuda-12.0/nv-tensorrt-local-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get install tensorrt
  • TensorRT
« 上一页
Linux:内核启动不了之后
下一页 »
开发实战:使用Spdlog管理调试信息
© 2025 AhaKnow Powered by Hugo & CKPaper