【教程】Install NVIDIA TensorRT on WSL 在WSL上安装英伟达TensorRT
NVIDIA® TensorRT™ 是一个用于高性能深度学习推理的 API 生态系统。TensorRT 包括推理运行时和模型优化,可为生产应用提供低延迟和高吞吐量。TensorRT 生态系统包括 TensorRT、TensorRT-LLM、TensorRT Model Optimizer 和 TensorRT Cloud。
本教程参照NVIDIA官网TensorRT安装教程展示如何在安装有WSL2 Ubuntu-22.04的windows11系统上安装英伟达TensorRT
I. 环境
- 具有NVIDIA GPU
- windows已安装NVIDIA driver
- 系统为windows11 安装了 WSL2 Ubuntu 22.04
- 理论上来说,如果windows上具有NVIDIA driver,wsl不需要再安装
- 通过nvidia-smi查看系统cuda版本,此处为12.4
II. 流程
本教程参照NVIDIA官网TensorRT安装教程编写
- 安装CUDA
- 安装cuDNN
- 安装TensorRT
III. CUDA安装
在NVIDIA官网TensorRT安装教程里选择与通过nvidia-smi中显示的一致的CUDA版本,本教程采用CUDA 12.4
本教程采用12.4.1版本