win10安装TensorRT(C++,Python)

前言

本偏文章主要用于记录Win10上TensorRT的安装过程。
C++软件:VS2019
python
PS:conda环境安装python依赖包时注意激活环境。

1.先决条件

在安装TensorRT前,首先需要安装CUDA、CUDNN等NVIDIA的基本库。
CUDA: 11.3; cmd执行nvcc -Vwin10安装TensorRT(C++,Python)

cudnn:8.2.1
win10安装TensorRT(C++,Python)
win10安装TensorRT(C++,Python)
py-cuda (我用的11.5)

pip install cuda-python==11.5
(以下包我也不清楚什么用,但是安装了)
pip install nvidia-pyindex
pip install graphsurgeon
pip install uff

二、TensorRT下载

1.官网下载

NVIDIA官网:下载链接
需要注册。
win10安装TensorRT(C++,Python)
选择tensorRT 8
win10安装TensorRT(C++,Python)
EA VS GA
win10安装TensorRT(C++,Python)
将下载的压缩包解压,然后放到自己合适的目录。如:D:\ProgramData\TensorRT-8.2.1.8

将解压后目录中的lib绝对路径添加到环境变量。

将lib目录下的dll文件复制到CUDA目录:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.3\bin。

2.C++

用VS2019 打开 TensorRT-7.2.2.3\samples\sampleMNIST\sample_mnist.sln
修改代码如图所示。
win10安装TensorRT(C++,Python)
右键sample_mnist项目,重新生成。win10安装TensorRT(C++,Python)
win10安装TensorRT(C++,Python)
打开D:\ProgramData\TensorRT-8.2.1.8\bin
win10安装TensorRT(C++,Python)双击sample_mnist.exe。
win10安装TensorRT(C++,Python)
你完成了!
但是,此时python是不能使用tensorRT的。

3.Python

控制台进入此目录。 win10安装TensorRT(C++,Python)
根据python版本安装whl。(activate py 为激活我的conda环境)
win10安装TensorRT(C++,Python)
做一个测试:
win10安装TensorRT(C++,Python)
完成的!

总结

没有总结,如果有用请点赞关注。
win10安装TensorRT(C++,Python)

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
乘风的头像乘风管理团队
上一篇 2022年3月21日 下午12:47
下一篇 2022年3月21日 下午1:07

相关推荐