win10安装TensorRT

下载TensorRT

根据自己的CUDA版本下载对应的TensorRT压缩包,我使用的NVIDIA GeForce 3090,CUDA11.1,使用nvcc -V查看CUDA版本

(base) C:\Users\Administrator>nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Mon_Oct_12_20:54:10_Pacific_Daylight_Time_2020
Cuda compilation tools, release 11.1, V11.1.105
Build cuda_11.1.relgpu_drvr455TC455_06.29190527_0

进入Nvidia官网下载对应的TensorRT版本,如下图所示:

win10安装TensorRT_第1张图片

解压缩

win10安装TensorRT_第2张图片
使用visual studio打开TensorRT下的samples\sampleMNIST工程
win10安装TensorRT_第3张图片
我使用的VS2017,打开后进行如下配置,右键→属性
win10安装TensorRT_第4张图片
由于TensorRT的lib文件夹下包含了dll文件和lib文件,因此我们如下配置 将TensorRT中的lib文件夹路径添加如下:VC++目录→可执行文件目录
win10安装TensorRT_第5张图片
将TensorRT中的lib文件夹路径添加如下:VC++目录→库目录
win10安装TensorRT_第6张图片
将TensorRT中的include文件夹路径添加如下:C/C++→常规→附加包含目录
win10安装TensorRT_第7张图片
将lib文件夹中的所有lib文件添加如下【nvinfer.lib、nvinfer_plugin.lib、nvonnxparser.lib、nvparsers.lib】:链接器→输入→附加依赖项
win10安装TensorRT_第8张图片
右键工程→重新生成
win10安装TensorRT_第9张图片

进入TensorRT中的bin文件夹,在cmd中运行sample_mnist.exe,出现如下所示则安装成功

win10安装TensorRT_第10张图片

你可能感兴趣的:(模型部署,深度学习)