程序员经验分享-hwhale

pytorch 的 dataset 中使用 onnxruntime

2023-11-09

如果在 dataset 中预处理图像时，用到了 onnxruntime 的 cudaep，出现这样的错误：

1、2022-12-13 13:53:01.554864883 [E:onnxruntime:Default, cuda_call.cc:119 CudaCall] CUDA failure 3: initialization error ; GPU=32765 ; hostname=ubuntu16 ; expr=cudaSetDevice(GetDeviceId()); 
2、onnxruntime.capi.onnxruntime_pybind11_state.Fail: Caught Fail in DataLoader worker process 0.
3、onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : CUDA error executing cudaSetDevice(GetDeviceId())
4、TypeError: cannot pickle 'onnxruntime.capi.onnxruntime_pybind11_state.InferenceSession' object

...

把 DataLoader 中 num_workers 设置成 0 就好了。

本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)

模式识别

编程

Pytorch

onnxruntime

numworkers

pytorch 的 dataset 中使用 onnxruntime 的相关文章

随机推荐

  • CMD设置代理 注册表设置IE代理

    法一 注册表reg 设置代理 开启代理 并 设置代理地址为127 0 0 1 8080 Windows Registry Editor Version 5 00 HKEY CURRENT USER Software Microsoft Wi

  • Vite的原理

    背景 这里的背景介绍会从与Vite紧密相关的两个概念的发展史说起 一个是JavaScript的模块化标准 另一个是前端构建工具 共存的模块化标准 为什么JavaScript会有多种共存的模块化标准 因为js在设计之初并没有模块化的概念 随着

  • 代码静态分析与安全检测工具COBOT

    COBOT是北大软件开发的软件代码静态分析与安全检测工具 COBOT支持10类1000余种编码规则检查 支持CWE 14类110余种语义缺陷 支持8类90余种常见的安全漏洞分析 COBOT通过度量分析能够检测包括圈复杂度 函数扇入扇出 注释

  • EMI滤波电路是由哪些元件组成的，一文看懂

    通常对于追求效率的电源来说 NTC热敏电阻几瓦的损耗始终会降低电源的转换效率 而且对于关机后在短时间内再次开机的情况 如果没有继电器 处于高温下的NTC热敏电阻将无法发挥正常作用 因此继电器与NTC在高端电源中往往是配套使用 以达到 鱼与熊

  • Macbook Pro M1芯片Python开发环境配置

    文章主要介绍M1 Mac新机器如何进行开发环境配置 由于在进行开发环境配置的时候 需要经常使用终端 而且新机器是没有配置homebrew的 后面一些操作不是很方便 所以本文将会对日常终端使用方面进行一些配置 丰富我们的终端样式和使用 文章大

  • Linux进阶_DNS服务和BIND之详解篇

    成功不易 加倍努力 1 名字解析介绍和DNS 1 1 DNS服务工作原理 1 2 DNS查询类型 1 3 名称服务器 1 4 解析类型 1 5 完整的查询请求经过的流程 2 DNS 服务相关概念和技术 2 1 DNS服务器的类型 2 2 区

  • Kafka学习资料全集，保证你从入门到精通

    1 Kafka架构和工作原理 https www jianshu com p aa4c6994687e 2 Prometheus Grafana kafka exporter监控kafka https blog csdn net an109

  • 【picodet 训练流程】

    picodet训练流程 一 Train 1 环境搭建 2 准备数据 2 1 数据位置 2 2 数据读取和标签读取 2 2 1yml 文件位置 2 2 2数据集路径文件生成 3 修改模型相应参数 3 1 参数文件位置 4 训练 二 Eval

  • window7系统中64位安装matalbR2009b后出现乱码的解决方案

    转自 http blog csdn net shaoguangleo archive 2010 11 29 6042194 aspx MATLAB 中默认的字体是 Monospaced 即等宽字体 这是一种非常适合用于显示程序源代码的字体

  • 使用Flex图表组件

    使用Flex图表组件 以图表或是图的方式显示数据的能力可以使得Flex程序用户的数据交互更为容易 与仅显示简单的数字数据表不同 我们可以显示条状图 饼图 线图或是其他类型的图表 并且可以使用颜色 标题以及二维图形来表示我们的数据 在这一部分

  • 求矢量夹角

    由点乘计算角度 由叉乘计算正负 点乘公式 a b a b cos

  • VQ-VAE

    Generating Diverse High Fidelity Images with VQ VAE 2 摘要 本文主要研究VQ VAE在大尺度图像生成器上的应用 我们改进了自回归的先验来生成比以前更加优秀的图片 本文使用的是简单的编码和

  • springbboot的默认缓存方案:@Cacheable@CachePut@CacheEvict

    Spring Boot 默认使用的缓存方案是基于 Spring Framework 的缓存抽象 在 Spring Boot 中 你可以使用 Cacheable CachePut 和 CacheEvict 注解来实现缓存功能 Cacheabl

  • Bugkuctf--逆向第三题游戏过关题解

    题目链接 https ctf bugku com challenges 先让我们来打开文件 既然题目都说是游戏过关了 那就玩玩这个游戏呗 雾 游戏还是很简单的 但显然这道题应该用逆向的思维来做 废话 正式部分 逆向的题拿到手一般能考虑的工具

  • 行走在前端路上

    从今天起开始写一写实际项目中遇到的各种前端问题 希望对自己和和小伙伴们都能起到一定的帮助作用

  • 架构师必须了解的 5 种最佳软件架构模式

    世界变得越来越依赖软件 软件系统已经渗透到了人类生活的方方面面 并带来了很多便利 从移动应用 用于和人联系 到医疗应用和深度学习模型 到金融技术系统 再到智能建筑 利用技术来自动化许多功能 为了提供所需的解决方案并获得最佳效果 必须使用恰当

  • Qt 疑云解答

    一 qt的mianwindow处不能布局 我们在mainwindow中添加2个控件 pushButton和pushButton 2 QWidget centerWindow new QWidget this this is point to

  • Hashmap扩容方法机制原理

    Hashmap扩容方法 机制原理 1 7版本 数组额定容量为16 元素数量超过负载因子 一般为0 75 后会扩容至原有数组大小 2 遍历老数组每个位置上的链表每个元素 取每个元素的key根据链表长度计算出在新数组中的下表 将元素添加到新数组

  • SylixOS下Redis编程实践

    1 Redis简介 Redis是一个完全开源的key value型数据库 其显著特点是基于内存操作 但是可以持久化到磁盘 相比较其他的数据库 Redis有着明显的优势 响应速度快和特性丰富是其明显的特点 其他的特点这里不一一赘述 Redis

  • pytorch 的 dataset 中使用 onnxruntime

    如果在 dataset 中预处理图像时 用到了 onnxruntime 的 cudaep 出现这样的错误 1 2022 12 13 13 53 01 554864883 E onnxruntime Default cuda call cc

热门标签