nvidia TensorRT在windows下安装并运行sampleMNIST示例

TensorRT在windows下安装并运行sampleMNIST示例

  1. 官网下载TensorRT对应版本,我选择TensorRT 6.0 GA for windows,下载网址https://developer.nvidia.com/nvidia-tensorrt-6x-download下载对应cuda10.1即可
  2. zip安装说明文档照着做就行,添加环境变量,复制仅dll文件等注意cuda10.1对应的cudnn为7.6.3,一定要对应上就行,https://docs.nvidia.com/deeplearning/tensorrt/archives/tensorrt-601/tensorrt-install-guide/index.html#installing-zip
  3. sampleMNIST注意,先下载好mnist数据集的4个文件,并解压,按照这个来,直接mnist官网下载也可以:https://github.com/NVIDIA/DIGITS/blob/master/docs/GettingStarted.md先下载目录DIGITS/digits/download_data的文件夹,然后cd到本地该文件夹处,推荐使用python2运行:
# python -m digits.download_data mnist /yourpath /mnist
python __main__.py mnist /yourpath/mnist

将gz文件解压所得的.idx3-ubyte放到/yourpath/TensorRT-6.0.1.5/data/mnist里面即可完成这一步骤
nvidia TensorRT在windows下安装并运行sampleMNIST示例_第1张图片
4. 修改该文件目录下的generate_pgms.py文件main函数中的parse中dataset的default为"train-images.idx3-ubyte"(原来的点为-)
5. 到/yourpath/TensorRT-6.0.1.5/samples/sampleMNIST下,VS双击打开sln文件,生成解决方案并运行之,得到如下结果即成功:
nvidia TensorRT在windows下安装并运行sampleMNIST示例_第2张图片

你可能感兴趣的:(TensorRT学习,深度学习,pytorch,cuda,windows)