ubuntu mmskeleton0.7版本与mmcv mmdetection适配安装

mmskeleton实验   

官方网站 https://github.com/open-mmlab/mmskeleton

官方安装说明    https://github.com/open-mmlab/mmskeleton/blob/master/doc/GETTING_STARTED.md

官网中提到

[2020-01-21] MMSkeleton v0.7 is released.

支持cuda 9.2 cuda 10.0 pytorch 1.2 python3.7

mmskeleton依赖两个重要的库,mmcv视频处理库,以及mmdetection目标检测库

官方网址 mmdetection mmcv

 之前实际上安装过mmskeleton,最近为了清理环境,想着重新装一次

电脑双cuda版本 cuda 9.0和 cuda 10.1

失败配置:

根据pytorch官网 pytorch各个版本安装说明安装cuda10.1支持的pytorch1.4

pip install mmcv

安装mmcv=1.0.5

编译mmdetection,安装 mmdet=2.3.0

在安装mmskeleton时,python setup.py develop成功安装mmskeleton 0.7

python mmskl.py pose_demo

demo验证时,开始报错。在debug过程中出现的错误如下

第一种,TypeError: init() got an unexpected keyword argument 'num_stages'

第二种,mmskeleton.utils.third_party attempted to use a functionality that requires module mmdet.apis, but it couldn't be loaded

第三种,  mmdet/ops/nms/src/nms_cuda.cpp:9:3: note: in expansion of macro ‘CHECK_CUDA’
       CHECK_CUDA(dets);
       ^

debug说明:

降低mmcv版本,pip install mmcv==0.4.3   mmdetection编译后仍然 为mmdet==2.3

mmcv-full,mmskeleton不支持

通过mmskeleton自动安装mmcv,mmdet失败

在mmskeleton setup.py文件中找到mmdetection=rc1版本下载网址手动下载

https://github.com/open-mmlab/mmdetection/archive/v1.0rc1.zip

cuda10.1 pytorch1.4 安装mmcv==0.4.3 mmdet==rc1+unknown无法导出mmdet cuda报错

成功配置:

cuda9.0 pytorch1.1 mmcv==0.4.3 mmdet==rc1+unknown

感言

特意升级了驱动,下载了cuda10.1,折腾两天,没用,o的k。

 

  • 1
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 5
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 5
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值