当前位置: 首页 > news >正文

TensortRT:sample.py:DeprecationWarning:

错误描述

sample.py:112: DeprecationWarning: Use set_memory_pool_limit instead.
config.max_workspace_size = common.GiB(1)
sample.py:75: DeprecationWarning: Use add_convolution_nd instead.
conv1 = network.add_convolution(
sample.py:78: DeprecationWarning: Use stride_nd instead.
conv1.stride = (1, 1)
sample.py:80: DeprecationWarning: Use add_pooling_nd instead.
pool1 = network.add_pooling(input=conv1.get_output(0), type=trt.PoolingType.MAX, window_size=(2, 2))
sample.py:81: DeprecationWarning: Use stride_nd instead.
pool1.stride = (2, 2)
sample.py:85: DeprecationWarning: Use add_convolution_nd instead.
conv2 = network.add_convolution(pool1.get_output(0), 50, (5, 5), conv2_w, conv2_b)
sample.py:86: DeprecationWarning: Use stride_nd instead.
conv2.stride = (1, 1)
sample.py:88: DeprecationWarning: Use add_pooling_nd instead.
pool2 = network.add_pooling(conv2.get_output(0), trt.PoolingType.MAX, (2, 2))
sample.py:89: DeprecationWarning: Use stride_nd instead.
pool2.stride = (2, 2)
在这里插入图片描述

原因:

这些警告信息是在运行Python代码时出现的,它们是关于TensorRT(一个用于深度学习推理的库)的API使用的。TensorRT的最新版本中对某些函数进行了更新,而代码中仍然使用了一些被弃用(deprecated)的函数。

解决方案:

config.max_workspace_size = common.GiB(1):

警告信息:DeprecationWarning: Use set_memory_pool_limit instead.
解决方案:使用 config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, common.GiB(1)) 来替代被弃用的 config.max_workspace_size。
在这里插入图片描述

config.set_memory_pool_limit(trt.MemoryPoolType.WORKSPACE, common.GiB(1))

network.add_convolution(…):

警告信息:DeprecationWarning: Use add_convolution_nd instead.
解决方案:使用 network.add_convolution_nd(…) 来替代 network.add_convolution(…)。
在这里插入图片描述

conv1.stride = (1, 1):

警告信息:DeprecationWarning: Use stride_nd instead.
解决方案:使用 conv1.stride_nd = (1, 1) 来替代 conv1.stride。
在这里插入图片描述

network.add_pooling(…):

警告信息:DeprecationWarning: Use add_pooling_nd instead.
解决方案:使用 network.add_pooling_nd(…) 来替代 network.add_pooling(…)。
在这里插入图片描述

pool1.stride = (2, 2)、pool2.stride = (2, 2):

警告信息:与 conv1.stride = (1, 1) 相同,同样需要使用 stride_nd。
在这里插入图片描述
再次运行python sample.py, 没有任何warning了。
在这里插入图片描述

感悟:

主要是我下载的TensorRT版本太新了,而sample.py里面有一些过时的TensorRT的语法,所以才导致今天的错误。

相关文章:

  • Linux shell编程学习笔记35:seq
  • 33.搜索旋转排序数组
  • Certbot实现 HTTPS 免费证书(Let‘s Encrypt)自动续期
  • 【Linux】tree命令使用
  • 第七章 SpringCloud Alibaba 实现微服务集成Sentinel
  • Ubuntu22,桌面服务状态查看与重启
  • 2024黑龙江省职业院校技能大赛暨国赛选拔赛“GZ031应用软件系统开发”赛项赛题题库
  • 3.DevEco Studio安装鸿蒙手机app本地模拟器
  • 2019年AMC8数学竞赛真题的典型考点和详细解析
  • 后端项目全局异常处理-使用RuntimeException自定义异常异常分类简单举例
  • 详细教程 - 从零开发 Vue 鸿蒙harmonyOS应用 第三节 (封装TabBar JS版)
  • 基于以太坊的智能合约开发Solidity(事件日志篇)
  • Java8 IfPresent 与 forEach 的组合操作
  • Nacos配置管理-微服务配置拉取
  • Linux Conda 安装 Jupyter
  • Github访问慢解决办法
  • IOS评论框不贴底(ios12新bug)
  • Java程序员幽默爆笑锦集
  • Java多态
  • Leetcode 27 Remove Element
  • mac修复ab及siege安装
  • nodejs调试方法
  • Python学习笔记 字符串拼接
  • socket.io+express实现聊天室的思考(三)
  • Vim Clutch | 面向脚踏板编程……
  • Webpack 4x 之路 ( 四 )
  • 简单数学运算程序(不定期更新)
  • 盘点那些不知名却常用的 Git 操作
  • 使用Maven插件构建SpringBoot项目,生成Docker镜像push到DockerHub上
  • 使用权重正则化较少模型过拟合
  • #图像处理
  • #我与Java虚拟机的故事#连载15:完整阅读的第一本技术书籍
  • #周末课堂# 【Linux + JVM + Mysql高级性能优化班】(火热报名中~~~)
  • (C++)栈的链式存储结构(出栈、入栈、判空、遍历、销毁)(数据结构与算法)
  • (done) NLP “bag-of-words“ 方法 (带有二元分类和多元分类两个例子)词袋模型、BoW
  • (带教程)商业版SEO关键词按天计费系统:关键词排名优化、代理服务、手机自适应及搭建教程
  • (详细版)Vary: Scaling up the Vision Vocabulary for Large Vision-Language Models
  • (一)UDP基本编程步骤
  • (转)IIS6 ASP 0251超过响应缓冲区限制错误的解决方法
  • (转贴)用VML开发工作流设计器 UCML.NET工作流管理系统
  • .bat批处理(一):@echo off
  • .md即markdown文件的基本常用编写语法
  • .NET gRPC 和RESTful简单对比
  • .NET I/O 学习笔记:对文件和目录进行解压缩操作
  • .NET 设计模式初探
  • .NET 设计模式—适配器模式(Adapter Pattern)
  • .NET/C# 反射的的性能数据,以及高性能开发建议(反射获取 Attribute 和反射调用方法)
  • .Net环境下的缓存技术介绍
  • .NET精简框架的“无法找到资源程序集”异常释疑
  • .sh
  • /usr/bin/env: node: No such file or directory
  • [ vulhub漏洞复现篇 ] ThinkPHP 5.0.23-Rce
  • [ 渗透工具篇 ] 一篇文章让你掌握神奇的shuize -- 信息收集自动化工具
  • [2013][note]通过石墨烯调谐用于开关、传感的动态可重构Fano超——
  • [C#]获取指定文件夹下的所有文件名(递归)