这个BUG我是在用Detectron2出现的,注意你安装的CUDA版本问题。
我安装的是CUDA 11.3,但是我在进行代码复现时,作者安装的是CUDA10.2的。
CUDA11.3对应的Detectron2的安装
python -m pip install detectron2 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html
这个BUG我是在用Detectron2出现的,注意你安装的CUDA版本问题。
我安装的是CUDA 11.3,但是我在进行代码复现时,作者安装的是CUDA10.2的。
CUDA11.3对应的Detectron2的安装
python -m pip install detectron2 -f \
https://dl.fbaipublicfiles.com/detectron2/wheels/cu113/torch1.10/index.html