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,为什么还会报错呢?有没有同学可以解答一下呢?

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

img

不知道你这个问题是否已经解决, 如果还没有解决的话:

如果你已经解决了该问题, 非常希望你能够分享一下解决方案, 写成博客, 将相关链接放在评论区, 以帮助更多的人 ^-^