detectron2 配置记录

时间:2019-12-11 19:32:05   收藏:0   阅读:483

 

 

 

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

 

原文:https://www.cnblogs.com/QingHuan/p/12024564.html

评论(0
© 2014 bubuko.com 版权所有 - 联系我们:wmxa8@hotmail.com
打开技术之扣,分享程序人生!