detectron2 配置记录
1. RuntimeError: Not compiled with GPU support (ROIAlign_forward at /home/oliver/PycharmProjects/detectron2/detectron2/layers/csrc/ROIAlign/ROIAlign.h:73)
先检查cuda版本和pytorch版本是否对应
然后重新安装:
python setup.py build --force develop
posted on 2019-12-11 19:18 Oliver-cs 阅读(1959) 评论(0) 编辑 收藏 举报