Advertisement

修复TensorRT安装中download_pgms.py程序出现错误的问题

阅读量：

问题描述

为提升yolov5模型的推理速度，可尝试借助TensorRT工具进行加速处理，其安装步骤在相关技术博客中已有详尽说明：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 o

全部评论 (0)

还没有任何评论哟~

相关文章推荐