在开始之前请确保已经安装好了CUDA和cuDNN
1.下载TensorRT
地址：https://developer.nvidia.com/nvidia-tensorrt-7x-download。按自己的CUDA版本下载，这里下载的是TensorRT-7.2.1
2. 将下载好的压缩包解压，找个位置放置
3.添加环境变量
4. 复制dll文件到cuda安装目录的bin文件夹下
5.进入TensorRT安装目录下的data/mnist，执行python脚本download_pgms.py下载数据集文件
下载完成后data/mnis下会多出几个.pgm文件
注：如果执行python download_pgms.py出现错误：”urllib.error.HTTPError: HTTP Error 503: Service Unavailable”，可以多试几次，或者从其他地方复制下载好的.pgm文件到data/mnist目录下即可：
6.使用Visual Studio打开samples/sampleMNIST/sample_mnist.sln，编译运行mnist示例程序，有类似如下结果则说明安装成功
评论 (0)