本文主要是介绍Detectron2在CPU上执行出现“ Torch not compiled with CUDA enabled”的错误,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
Detectron2在CPU上执行出现“ Torch not compiled with CUDA enabled”的错误
在没有GPU显卡的电脑上配置Detectron2环境,配置OK后,运行如下代码,权重文件提前下载好,放在一个固定的位置,出现错误“Torch not compiled with CUDA enabled”
python demo/demo.py \--config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \--input img/test1.jpg --output outputs/ \--opts MODEL.WEIGHTS '/home/project/detectron2/pre_train_model/model_final_f10217.pkl' \
错误如下图:
解决方法:在opts后面跟上 MODEL.DEVICE cpu
python demo/demo.py \--config-file configs/COCO-InstanceSegmentation/mask_rcnn_R_50_FPN_3x.yaml \--input img/test1.jpg --output outputs/ \--opts MODEL.WEIGHTS '/home/project/detectron2/pre_train_model/model_final_f10217.pkl' \MODEL.DEVICE cpu
在GPU电脑上配置detectron2可参考Detectron2安装测试
这篇关于Detectron2在CPU上执行出现“ Torch not compiled with CUDA enabled”的错误的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!