Detectron2在CPU上执行出现“ Torch not compiled with CUDA enabled”的错误

最新推荐文章于 2024-07-02 09:44:59 发布
最新推荐文章于 2024-07-02 09:44:59 发布
阅读量2.8w 收藏 17
点赞数 5
分类专栏： Detectron2 PyTorch 文章标签： Detectron2 CUDA
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/happyday_d/article/details/103330155
版权

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安装测试

NH558J0
关注
  • 5
    点赞
  • 踩
  • 17
    收藏
    觉得还不错? 一键收藏
  • 6
    评论
专栏目录
【Detectron2】代码库学习-3. 代码结构
wendell 的博客
11-14 783
Detectron2是Facebook AI Research(FAIR)推出的基于Pytorch的视觉算法开源框架，主要聚焦于目标检测和分割任务等视觉算法，此外还支持全景分割，关键点检测，旋转框检测等任务。Detectron2继承自Detectron 和mask-rcnn。Detectron2具有较强的灵活性和可扩展性，支持快速的单GPU训练，多GPU并行训练和多节点分布式训练。
pytorch-faster-rcnn：pytorch1.0已更新。 支持CPU测试和演示。 （使用detectron2，这是一个杰作）
02-22
公告（2019.11.2） 此仓库是两年前建立的，当时还没有可以实现合理性能的pytorch检测实现。 目前，有许多更好的存储库，例如： （这是一个杰作。） 因此，将不会积极维护此仓库。 重要通知： 如果您在2017年9月26日之前使用master分支及其相应的预训练模型，请注意：现在，旧的master分支位于old_master下，您仍然可以运行代码并下载预训练的模型，但是该旧master的预训练模型与当前的主站不兼容！ 新老master分支之间的主要区别在于这两个提交： ， 更改与该； master现在匹配中的所有细节，因此我们现在可以将预训练的tf模型转换为pytorch模型。 pytorch-faster-rcnn 基于Chenxinlei Chen的的pyrc的快速RCNN检测框架。 Chen Xinlei Chen的存储库基于提供的更快RCNN的python Ca
detectron2_predict_multigpu:Detectron2预测视频脚本
04-07
概述 预测大型视频数据集上的目标检测。 它将作业拆分为多个GPU，但请注意，它不使用多个GPU来预测单个图像。 例如，如果您有4个GPU，则必须打开4个终端并像这样运行。 CUDA_VISIBLE_DEVICES=0 python main.py --frames-input-dir " /path/to/input/frames " --output " /path/to/output " --divide-job-count 4 --divide-job-index 0 CUDA_VISIBLE_DEVICES=1 python main.py --frames-input-dir " /path/to/input/frames " --output " /path/to/output " --divide-job-count 4 --divide-job-index 1 CUDA_
Windows11+cpu 配置detectron2（超超超保姆级教学）
ty131456的博客
03-26 1091
window11系统下配置detectron2环境，并用cpu测试单个图片。
Torch not compiled with CUDA enabled 最新解决办法 pytorch 报错解决，踩坑得救！
最新发布
amelias的博客
07-02 2138
这个时候笔者就要问各位配置环境的小可爱们，要怎么选择pytorch的版本，有些可爱们直接可能就选cuda的12.4版本的，有些可爱谨慎一些，觉得不要选那么新的，那就选个12.1版本的，觉得应该可以兼容（这个小逻辑是正确的），运气好的话，这样是可以通过的。报错非常的头疼，尝试各种方法，都失败，花了三天时间终于将原因搞清楚，访问了大量的网站平台和博客，尝试都失败了，就在笔者准备全部重新装cuda和cudann的时候（现在证明装了也还是一样的不能解决问题），笔者将此分享，希望能帮助到需要帮助的可爱们。
Detectron2安装测试
热门推荐
happyday_d的博客
12-10 2万+
Detectron2安装测试 Detectron2是FAIR开源的基于Pytorch1.3.1的目标检测算法实现. Detectron2-github 1.Detectron2安装 1.1 安装配置基本环境 可参考项目中的Installtion conda create -n detectron2 python=3.7 conda activate detectron2 PyTorch ...
【detectron2】总结并完成一个道路积水的分割任务带数据集 cpu+gpu版本
weixin_40293999的博客
10-17 2335
mask-rcnn 实例分割 全景分割 安装使用指引，带数据集和最终效果，用detectron2家族实现，包括遇见的坑点和解决方案
【小白学习记录4】安装opencv，detectron2并在CPU上运行
weixin_49235359的博客
06-30 934
激活记录2中创建的pytorch虚拟环境，然后输入 pip install opencv-python 显示成功安装后，输入python import cv2 无报错
detectron2源码_conda包_torch包下载地址.txt
04-25
由于文件较大，里面提供了下载地址。包含内容：detectron2代码+conda依赖包+测试模型+torch包。代码已经配置好，配置好环境应该直接能运行
torch-1.14.0+cuda11.8+sm-90+deepspeed
06-03
nvidia新卡h100支持算力90，torch官网提供的下载torch-1.13只支持到86，torch-2.0和torch-2.1倒是支持90，但deepspeed又不支持torch2 无奈下我编译了torch-1.14，配合cuda11.8，能支持h100运行在deepspeed下训练大...
pytorch查看torch.Tensor和model是否在CUDA上的实例
09-18
首先，我们可以导入必要的库，包括`torch`和`torchvision.models`，然后创建一个简单的模型，例如VGG11，并检查模型的第一个参数是否在CUDA设备上： ```python import torch import torchvision.models as models ...
detectron2依赖包.zip
11-07
安装Detectron2必要的环境依赖，适用Linux环境，python3.7，参考csdn链接https://blog.csdn.net/u013347145/article/details/102945946
Detecron2源码阅读-训练器（3）
sqwev的博客
02-23 948
一个有意思的问题，DefaultTrainer继承的是TrainerBase，而不是SimpleTrainer。我们着重研究一下Trainer怎么写，还有多GPU训练的问题。还有一个想吐槽的点是detectron2全程通过cfg来配置，但是也没教俺们怎么写config。tips：config的说明在 config/defaults.py里。
自学Detectron2
weixin_46421825的博客
11-11 1223
官方文档：作为初学者，我反正是大多不能理解Getting Started with Detectron2 — detectron2 0.6 documentationhttps://detectron2.readthedocs.io/en/latest/tutorials/getting_started.html安装：我是在武大超算上安装的（Linux），python=3.7，pytorch=1.8，CUDA=10.2，这样一个虚拟环境，特别要注意的要安装一下依赖包后，才能正常装好detectron2:
Detectron2-入门1-极链AI云平台部署-多图好玩测试
weixin_41793473的博客
07-07 682
Detectron2：安装与测试 作者是根据Detectron2的官方文档来学习哒，但是到了实际应用上也会有些许差别。也许有很多bug，也许一帆风顺。 动手实践才是真理。一起加油吧！ Detectron2：GitHub地址：https://github.com/facebookresearch/detectron2 Detectron2官网文档：https://detectron2.readthedocs.io/en/latest/tutorials/index.html 选择部署平台 极链AI云平台 优势
检测数据库detectron2代码解析（一）训练文件
蓝鲸鱼的博客
08-06 1951
facebookresearch发表的检测数据库detectron2几乎涵盖了当下最新的各种检测代码。 训练代码 train_net.py 1. 参数加载 args = default_argument_parser().parse_args() 例如，在训练时传入参数： python tools/train_net.py --config-file configs/COCO-Detection/faster_rcnn_R_50_FPN_1x.yaml \ --num-gpus 1
Detectron2的入门学习
Lonery的博客
05-02 1万+
文章目录前言一、Detectron2的安装二、简单的运行案例1.利用已有的模型进行各种测试2.训练自己的模型总结 前言   detectron2是Facebook的一个机器视觉相关的库，建立在Detectron和maskrcnn-benchmark基础之上，可以进行目标检测、语义分割、全景分割，以及人体体姿骨干的识别。这个项目在GitHub上已经有超过了20k的星星。 一、Detectron2的安装   在Detectron2的官网上已经给出了Linux平台的安装方法。在这里介绍另外一个常用的平台即wi
GPU安装detectron2 构建踩坑日记
scl52tg的博客
03-03 682
安装detectron2自述经历安装条件安装步骤新建anaconda环境下载opencv-python拉取detectron2包构建错误相应集合,之后更新错误类型visual studio2019pytorch 与cuda版本问题 自述经历 在windows下配置这些环境什么的 报的错都是特别离谱的？？？网上搜也搜不到！ 本人还是建议有条件的还是使用linux系统吧！！！！！ 安装条件 需要配置好相应版本的NVIDIA驱动、CUDA、Pytorch、visual studio，visual stdio20
torch版本1.7.1+cpu出现AssertionError: Torch not compiled with CUDA enabled
05-05
当你在使用torch版本1.7.1+cpu时，如果出现了AssertionError: Torch not compiled with CUDA enabled的错误，这通常是因为你的torch版本没有启用CUDA支持。 CUDA是NVIDIA提供的并行计算平台和编程模型，它可以利用...
评论 6
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值