1、TensorRT学习笔记之安装TensorRT

目录

        1、查看cuda版本

        2、下载TensorRT

        3、安装

        4、环境变量配置

        5、验证

        6、遇见的问题

---

1、查看cuda版本

输入：nvidia-smi查看cuda版本

如同，我的cuda版本是12.1

nvidia-smi

2、下载TensorRT

根据cuda版本下载对应的tensorrt版本。

下载地址：NVIDIA TensorRT 下载 | NVIDIA 开发者

3、安装

• cmd进入到tensort所在目录的python文件，
• 输入pip install tensorrt-8.4.1.5-cp39-none-win_amd64.whl
• 画框的是python版本，在当前目录下打开cmd，直接输入python 加对应的文件名

4、环境变量配置

• 将lib文件添加进环境变量中
• 我的电脑
• 属性
• 高级系统设置
• 环境变量
• 系统变量中的path
• 新建：将tensorrt路径添加进去

5、验证

        成功输出trt版本和hello trt!!说明验证成功

import tensorrt as trt

if __name__ == "__main__":
    print(trt.__version__)
    print("hello trt!!")

6、遇见的问题

报错：Could not find: nvinfer.dll

解决方法：将把tensorRT里面的bin、include、lib复制到CUDA安装同名目录里面。（前提是cuda安装是正确且能运行的）

下一篇：2、TensorRT学习笔记之PT转ONNX、可视化ONNX