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版本,如下图所示:

解压缩

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

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

猜你喜欢

转载自blog.csdn.net/qq_39567427/article/details/122242456