win10安装TensorRT

2.1 下载TensorRT

版本使用TensorRT 7.2.3 for Windows，CUDA版本是11.0，

下载地址：NVIDIA TensorRT 7.x Download | NVIDIA Developer

文件大小约500M，

2.2 配置环境变量

将下载好的文件赋值到C盘根目录，然后将“C:\TensorRT-7.2.3.4\lib”添加环境变量。如下图：

2.3 安装必要的whl文件

在shell里面安装下面三个whl文件。执行命令：

 pip install nvidia-pyindex
 pip install graphsurgeon
 pip install uff

  

 

  
1
  
2
  
3
 

2.4 复制dll文件到cuda安装目录

复制到：C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0\bin，如下图：

到这里就安装完成了，下面我们测试一下。

2.5 测试

使用VS2017，打开C:\TensorRT-7.2.3.4\samples\sampleMNIST的sample_mnist.sln，如图：

然后右键sample_mnist项目，右键选择属性。然后再属性页选择“VC++目录——可执行文件目录”，打开“可执行文件目录”，将“C:\TensorRT-7.2.3.4\lib”添加进去，然后点击“确定”。

选择“C/C++——附加包含目录，点进去添加“C:\TensorRT-7.2.3.4\bin”，然后点击“确定”。

右键“sample_mnist”工程，然后选择“重新生成”。

看到如下信息，则表明生成成功。

生成后的文件如下图：

然后执行在data-》mnist下面运行“download_pgms.py”文件，下载数据集。

然后执行调试。如下图

出现下图则表明成功。

文章来源: wanghao.blog.csdn.net，作者：AI浩，版权归原作者所有，如需转载，请联系作者。

原文链接：wanghao.blog.csdn.net/article/details/120360288