【MMDetection系列】遇到的BUG汇总


前言

记录在学习使用mmdetection框架时,遇到的一些bug,随时补充。


本文初始环境配置为:
GPU:NVIDIA GeForce RTX 3080 (arch=8.6)
cuda:10.1
Pytorch:1.8.1
torchvision:0.9.1+cu101
torchaudio:0.8.1


一、使用MMDetection已有模型在标准数据集上进行推理时遇到的问题

问题来源:

在学习使用MMdetection已有模型在标准数据集上进行推理时,使用 DetInferencer来获得推理结果,运行代码如下:

from mmdet.apis import DetInferencer

# 初始化模型
inferencer = DetInferencer('rtmdet_tiny_8xb32-300e_coco')

# 推理示例图片
inferencer('demo/demo.jpg', show=True)

结果出现了错误信息1:

GeForce RTX 3080 with CUDA capability sm_86 is not compatible with the current PyTorch installation.

在这里插入图片描述可参考:https://blog.youkuaiyun.com/weixin_43751285/article/details/110651105

以及错误信息2:

RuntimeError: CUDA error: no kernel image is available for execution on the device

在解决如上两个问题的过程中,接连遇到一些错误提示信息如:

ImportError: libcudart.so.10.1: canno

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值