解决linux虚拟环境上的AttributeError: module ‘torch.utils.data’ has no attribute ‘collate’问题

在运行LPRnet时遇到torch.utils.data缺失collate包的错误。解决方案是在虚拟环境的torch.utils.data目录下,将_utils目录中的collate.py文件复制到data目录中,以此解决依赖问题。此方法适用于不同Python版本的虚拟环境。
摘要由CSDN通过智能技术生成

今天在另一台电脑中首次跑LPRnet出现了一个这样的错误, torch.utils.data 下没有collate这个包。
在这里插入图片描述
因为是在虚拟环境中运行,所以记录一下在虚拟环境中怎么解决这样的问题。
虚拟环境的torch.utils.data路径到底在哪?如图
在这里插入图片描述
“learn”是自己的环境名(没有和第一张图对应上,当时截图没注意,如果和第一张图对上那么就应该是“cch”),python3.8是创建虚拟环境时的python版本,如果创建的是3.7版本那对应的目录就是python3.7。
将_utils目录下的collate.py复制到上一级data目录中,问题解决。
在这里插入图片描述

评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值