机器环境:ubuntu16.04,cuda8.0
1.torch&tensorflow 版本
源码地址:https://github.com/MVIG-SJTU/AlphaPose
首先跑的是torch版本,安装了torch和tensorflow后运行程序出现如下错误:
Error in `python2': corrupted size vs. prev_size: 0x0000000002981e70
可见issue"https://github.com/MVIG-SJTU/AlphaPose/issues/88"
放弃该版本,根据作者建议转向pytorch版本。
2.pytorch版本
首先安装pytorch,源码使用的是python3,我的机器系统是ubuntu16.04,自带python3.5,因此首先安装python3.5对应的pytorch。pytorch官网:https://pytorch.org/。因为网络问题,需要选择之前版本下载,如下图所示:
选择python3.5 和 cuda 8.0对应的版本:
安装之后,按照源码后续步骤下载预训练模型即可。
# pytorch 版本
-- pytorch 1.0.1
-- torchvision 0.2.1
https://github.com/MVIG-SJTU/AlphaPose/issues/182