流氓兔欸 2023-04-14 20:55 采纳率: 0%
浏览 285

yolov7+DCNv3 报错:RuntimeError: Not implemented on the CPU

File "/home/undergraduate/anaconda3/envs/yolov7p/lib/python3.8/site-packages/torch/cuda/amp/autocast_mode.py", line 110, in decorate_fwd
    return fwd(*args, **kwargs)
  File "/media/kb541/ruanpuchao/yolov7_train/models/ops_dcnv3/functions/dcnv3_func.py", line 39, in forward
    output = DCNv3.dcnv3_forward(
RuntimeError: Not implemented on the CPU

我在yolov7中添加可变形卷积DCNv3后,在训练时报错:无法应用到CPU,但是我的device是cuda:0,1,为什么还会报错呢?有没有同学可以解答一下呢?

  • 写回答

2条回答 默认 最新

  • 栗子甜酒 2023-06-17 11:13
    关注

    我也遇到了这个问题,不知道up解决了吗

    img

    评论

报告相同问题?

问题事件

  • 创建了问题 4月14日