win10安装TensorRT

Posted 2021-10-09 AI浩

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

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

然后执行调试。如下图

出现下图则表明成功。