【Win10安装Tensorrt和torch2trt】

Win10安装Tensorrt和torch2trt

  • [1 前言](#1 前言)
    • [1.1 实验环境](#1.1 实验环境)
    • [1.2 下载链接](#1.2 下载链接)
  • [2 具体安装](#2 具体安装)
    • [2.1 TensorRT安装](#2.1 TensorRT安装)
      • [2.1.1 设置环境变量](#2.1.1 设置环境变量)
      • [2.1.2 安装TensorRT资源包中的3个whl文件](#2.1.2 安装TensorRT资源包中的3个whl文件)
      • [2.1.3 复制tensorrt的库到CUDA安装目录的指定文件夹](#2.1.3 复制tensorrt的库到CUDA安装目录的指定文件夹)
    • [2.2 安装torch2trt](#2.2 安装torch2trt)
    • [2.3 验证是否安装成功](#2.3 验证是否安装成功)

1 前言

1.1 实验环境

(1)主机系统:Win10

(2)显卡: Quadro RTX 6000

(3)CUDA版本:V11.7.64

(4)Python:3.8.18

(5)TensorRT:8.6.1.6

1.2 下载链接

(1)TensorRT8.6.1.6下载链接(下载资源包): https://developer.nvidia.com/tensorrt

(2)pycuda下载链接((pip 离线安装):https://pypi.org/project/pycuda/2022.1/#history

也可在线下载:pip install pycuda;用于从 Python 访问NvidiaCUDA并行计算 API。

(3)torch2trt下载链接((pip 离线安装):https://github.com/NVIDIA-AI-IOT/torch2trt

(4)在线安装onnx和cython: pip install onnx cython

2 具体安装

2.1 TensorRT安装

2.1.1 设置环境变量

1、 解压下载的资源包,并将TensorRT对应的lib位置添加至系统环境变量。

2.1.2 安装TensorRT资源包中的3个whl文件

如上图所示,分别在graphsurgeon、python、uff在3个文件夹中找到对应的以下3个whl文件,并使用pip install 进行安装。

python 复制代码
pip install graphsurgeon-0.4.6-py2.py3-none-any.whl
pip install tensorrt-8.6.1-cp38-none-win_amd64.whl
pip install uff-0.6.9-py2.py3-none-any.whl

2.1.3 复制tensorrt的库到CUDA安装目录的指定文件夹

(1)将 TensorRT-8.6.1.6\include中头文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\include

(2)将TensorRT-8.6.1.6\lib 中所有lib文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\lib\x64

(3)将TensorRT-8.6.1.6\lib 中所有dll文件复制到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\bin

2.2 安装torch2trt

bash 复制代码
git clone https://github.com/NVIDIA-AI-IOT/torch2trt.git
cd torch2trt
python setup.py install

2.3 验证是否安装成功

不报错即表示安装成功!

相关推荐
shut up43 分钟前
LangChain - 如何使用阿里云百炼平台的Qwen-plus模型构建一个桌面文件查询AI助手 - 超详细
人工智能·python·langchain·智能体
宝贝儿好1 小时前
【python】第五章:python-GUI编程
python·pyqt
闲人编程2 小时前
从多个数据源(CSV, Excel, SQL)自动整合数据
python·mysql·数据分析·csv·存储·数据源·codecapsule
B站_计算机毕业设计之家2 小时前
推荐系统实战:python新能源汽车智能推荐(两种协同过滤+Django 全栈项目 源码)计算机专业✅
大数据·python·django·汽车·推荐系统·新能源·新能源汽车
茯苓gao2 小时前
Django网站开发记录(一)配置Mniconda,Python虚拟环境,配置Django
后端·python·django
Full Stack Developme2 小时前
Python Redis 教程
开发语言·redis·python
码界筑梦坊2 小时前
267-基于Django的携程酒店数据分析推荐系统
python·数据分析·django·毕业设计·echarts
Cherry Zack2 小时前
Django视图进阶:快捷函数、装饰器与请求响应
后端·python·django
qq_4924484463 小时前
Jmeter设置负载阶梯式压测场景(详解教程)
开发语言·python·jmeter
lianyinghhh3 小时前
瓦力机器人-舵机控制(基于树莓派5)
人工智能·python·自然语言处理·硬件工程