随着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
py-cuda 选择 11.5，安装如下

pip install cuda-python==11.5

二、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 trtif __name__ == "__main__":print(trt.__version__)print("hello trt!!")

TensorRT(10):python版本安装相关推荐

anaconda降级python失败_如何降级Python版本安装spyder?
如何降级 Python 版本安装 Spyder? 可能并不需要降级 Python 版本,使用 Spyder 独立安装包安装,然后指定 Python 解释器:另外一种方法是在 Anaconda 中安装. ...
Windows解决anaconda下双python版本安装TensorFlow
首先,就是双版本anaconda的安装: 以前安装好的是python2.7版本,而TensorFlow的安装仅支持3.5版本的.但是自己本来的2.7版本又不想遗弃.所以安装双版本的: 在anacond ...
GDAL的python版本安装使用
首先,编译Python版本的GDAL,具体编译可以参考之前的博客GDAL之Swig编译里面的内容,网址是:http://blog.csdn.net/liminlu0314/article/detail ...
python版本安装
目的本文目的在于,对于不熟悉Python的人,教你: 1. 从哪里找到可以下载到各种版本的包括Python 2.x和Python 3.x的最新版本的 Python. 高手请无视之. 2.以及 ...
指定python 版本安装相关插件
2019独角兽企业重金招聘Python工程师标准>>> 1. linux系统在linux安装了多版本python时(例如python2.6和2.7),pip安装的包不一定是用户想 ...
Mysql 5.7.10以上版本安装大坑
mysql解压缩版的配置已经方便无比了,但是也正是由于官方的不断优化,导致传统的套路一次次被修改.也让像我这样的萌新撞了个大墙. [注:本篇博客适用mysql5.7.10~5.7.15,如果版本已太过 ...
给指定Python版本安装指定版本库的办法
pip3.5 tensorflow==1.4 给Python3.5装TF1.4版本库 pip3.7 numpy 给Python3.7装numpy库直接调用pip3.5安装库appledeMBP:~ ...
mysql for mac 10.12_mac 10.12 版本安装mysql - sylvia的博客
今天安装mysql 遇到了坑,总结如下: 我的mac版本是10.12.1,于是安装了 https://dev.mysql.com/downloads/mysql/ 下的DMG Archive ...
Zeppelin(0.10.1版本)安装及创建hive解释器
目录 1. Zeppelin安装 2. Zeppelin网页配置连接hive 利用Zeppelin工具操作hive,需确保电脑已经启动Hadoop集群和hiveserver2服务. 在master主 ...

TensorRT(10):python版本安装

一、底层库依赖

二、windows下安装TensorRT

二、Linux下安装

三、测试是否安装成功

TensorRT(10):python版本安装相关推荐

最新文章

热门文章