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”文件，下载数据集。

然后执行调试。如下图

出现下图则表明成功。

Windows

版权声明：本文内容由网络用户投稿，版权归原作者所有，本站不拥有其著作权，亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容，请联系我们jiasou666@gmail.com 处理，核实后本网站将在24小时内删除侵权内容。