解决“测试TensorRT安装过程中download_pgms.py报错”问题

Zpadger

已于 2022-10-24 15:06:13 修改

阅读量1k

点赞数 1

分类专栏： Python debug 踩坑记录文章标签： python ubuntu TensorRT 1024程序员节

于 2021-05-06 11:42:51 首次发布

本文链接：https://blog.csdn.net/SpadgerZ/article/details/116450345

版权

在Ubuntu18.04上安装TensorRT过程中遇到download_pgms.py脚本错误，包括函数命名错误和网络连接问题。解决方法包括修改函数名和多次尝试以克服网络不稳定导致的下载失败。通过这些步骤，最终成功验证了TensorRT的安装。

摘要由CSDN通过智能技术生成

问题描述

尝试使用TensorRT对yolov5模型推理加速，首先进行安装，参考博客已经写的很详细了：Ubuntu1804+CUDA10.0安装TensorRT7

对了，在更改环境配置的时候还添加了CUDA和cuDNN：

vim ~/.bashrc

# tensorrt cuda and cudnn
export CUDA_INSTALL_DIR=/usr/local/cuda
export CUDNN_INSTALL_DIR=/usr/local/cuda


source ~/.bashrc

但是在测试TensorRT是否安装成功时，需要下载MNIST数据集，运行download_pgms.py会有两个报错：

1.函数报错

download_pgms.py:12: DeprecationWarning: The binary mode of fromstring is deprecated, as it behaves surprisingly on unicode inputs. Use frombuffer instead
  raw_buf = np.fromstring(buffer, dtype=np.uint8)
download_pgms.py:23: DeprecationWarning: The binary mode of fromstring is deprecated, as it behaves surprisingly on unicode inputs. Use frombuffer instead
  raw_buf = np.fromstri