标签：__ python TensorRT tensorrt 版本 安装

随着TensorRT8.0版本的发布，windows下也正式支持Python版本了，跟紧NVIDIA的步伐，正式总结一份TensorRT-python的使用经验。

一、底层库依赖

在安装TensorRT前，首先需要安装CUDA、CUDNN等NVIDIA的基本库，如何安装，已经老生常谈了，这里不再过多描述。
关于版本的选择，楼主这里：

CUDA版本，楼主这里选择的是 cuda11.5 ，具体cuda版本见 https://developer.nvidia.com/cuda-toolkit-archive，可自行下载。
CUDNN版本，选择 cudnn-11.3-windows-x64-v8.2.1.32，官网下载需要先注册账号，https://developer.nvidia.com/rdp/cudnn-archive

二、windows下安装TensorRT

首先去官网下载对应的TensorRT版本
https://developer.nvidia.com/nvidia-tensorrt-8x-download
楼主选的这，再选择对应的系统版本

解压下来对应多个文件，把lib里的所有dll库都拷贝到cuda在的目录下(或者添加到环节变量)。

拷贝到 C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.5\bin

最后，控制台进入安装包的python目录，选择对应的python版本进行安装即可

如楼主这里是python3.8
则在控制台下CD到当前目录，安装即可

pip install tensorrt-8.2.3.0-cp38-none-win_amd64.whl

二、Linux下安装

linux安装步骤和windows下是一样的，只不过需要选择对应的下载包，需要注意一下，ubuntu和Centos是不同的安装包

三、测试是否安装成功

写个脚本就可以啦

建立一个python工程，测试是否安装成功

```python
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
"""                  
*  * *** *  * *  *      
*  *  *   **  *  *             
****  *   **  *  *                 
*  *  *   **  *  *         
*  * **  *  * ****                

@File    : hello_tensorrt.py
@Date    : 2022/2/10/010
@Require :
@Author  : https://blog.csdn.net/hjxu2016
@Funtion : 

"""
import tensorrt as trt

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

标签：__,python,TensorRT,tensorrt,版本,安装
来源： https://blog.csdn.net/hjxu2016/article/details/122868139

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。