[踩坑全记录]CornerNet-Lite

这篇博客记录了在尝试运行CornerNet-Lite时遇到的常见错误,包括ModuleNotFoundError: No module named 'top_pool',找不到NVIDIA设备,invalid device ordinal错误,CUDA内存不足,以及RuntimeError: cannot join current thread。解决方法包括重建环境,调整CornerNet.json配置文件,以及修改tqdm包中的相应代码。
摘要由CSDN通过智能技术生成

运行demo.py时报错ModuleNotFoundError: No module named ‘top_pool’

原因:先compling nms等,然后才conda建立的环境
解决:建好环境后重新clone项目,重新编译

运行demo.py时找不到navida device

原因:可能是pytorch版本的问题
解决:放弃了用自己的环境,还是老老实实按照作者的conda-list重新建立了环境

invalid device ordinal at xxxxx

原因:gpu调用问题
解决:修改<CornerNet_Lite>/configs/CornerNet.json 里的 chunk_sizes,len改成有的gpu数目

CUDA out of memory

原因:gpu超memory了
解决:修改<CornerNet_Lite>/configs/CornerNet.json 里的 chunk_sizes,调小一点。同时记得修改batch_szie(等于sum(chunk_sizes))

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值