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。