本文主要是介绍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”的错误的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！
