Win10 安装 Tensorrt和torch2trt教程

本文详细介绍了在Windows 10系统中,使用NVIDIA GeForce RTX 3090显卡,CUDA 11.6和Python 3.9环境下,如何安装和配置TensorRT 8.5.1.7以及torch2trt。首先确保已安装CUDA和cuDNN,然后下载所需资源,通过pip安装ONNX和cython。接着,离线安装TensorRT及其依赖,并将库文件复制到CUDA安装目录。最后,通过离线安装方法设置torch2trt,并验证安装成功。遇到推理速度慢的问题时,检查CUDA和TensorRT版本的匹配性。

说明:实测有效。

1 电脑环境

系统:Win10
显卡:NVIDIA GeForce RTX 3090
cuda版本:CUDA 11.6.0 (后更新到CUDA 11.6.2)
python版本:3.9
TensorRT安装版本:8.5.1.7
(TensorRT与CUDA版本匹配,参见文献[4][5])

2 下载安装库

2.1 所需资源(假定已经安装cuda和cudnn)

1)TensorRT-8.5.1.7(下载资源包,下载链接:https://developer.nvidia.com/tensorrt
2) ONNX(pip 在线安装)
3) cython(pip 在线安装)
4) pycuda(pip 离线安装,下载链接:https

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值