windows10下面安装alphapose解决 ImportError : cannot import name ‘deform_conv_cuda‘

0 环境

conda create -n alphapose python=3.6

source activate alphapose

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=10.0 cudnn=7.6.5

系统：windows10

显卡：GTX 1070 8G

cudatoolkit: 10.0

torch: 1.1.0

torchvision: 0.3.0

cudnn: 7.6.5

在windows下面编译alphapose进行人体姿态估计工程时，执行：

python setup.py build develop

有些依赖项安装时间很长，可以停掉后使用pip手动安装即可，每次安装一次就 rm -rf build, 重新执行：

python setup.py build develop

出现上图就是安装成功了，如果没有安装成功则多执行几次，直到安装成功。

1. 运行demo

安装成功后运行demo，首先下载模型并放到正确的位置，如下图所示：

接着执行：

 python scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml --checkpoint pretrained_models/fast_res50_256x192.pth --indir examples/demo/ --outdir alphapose_results/ --save_img

出现错误：ImportError : cannot import name 'deform_conv_cuda'

于是去了alphapose的GitHub的issue中，发现一个解决办法，如下图所示：

于是我打开alphapose/setup.py，果然在第124行发现：force_compile = False

于是我将 force_compile = False 修改为 force_compile = True，然后执行：

rm -rf build
python setup.py build develop

再次运行demo:

 python scripts/demo_inference.py --cfg configs/coco/resnet/256x192_res50_lr1e-3_1x.yaml --checkpoint pretrained_models/fast_res50_256x192.pth --indir examples/demo/ --outdir alphapose_results/ --save_img

便成功了

alphapose_result下面的结果:

2.参考资料

2.1  【AlphaPose】AlphaPose运行环境搭建与测试

2.2  cannot import name 'deform_conv_cuda' from 'alphapose.models.layers.dcn #730