ST-GCN docker 安装

折腾了两天,在docker容器里安装过旧版ST-GCN和新版ST-GCN,新版成功了,旧版还没有折腾出来,这里也放出来,如果有成功的小伙伴可以来讨论一波~

一、成功案例(新版STGCN)——简单,官方给提供dockerfile

代码:https://github.com/open-mmlab/mmskeleton

 

1.使用的镜像:mmdetection创建容器

https://github.com/open-mmlab/mmdetection/blob/master/docs/INSTALL.md

docker build -t mmdetection docker/sudo docker run -it --runtime=nvidia  -v /tmp/.X11-unix:/tmp/.X11-unix -e DISPLAY=unix$DISPLAY -e GDK_SCALE -e GDK_DPI_SCALE -v /home/lxy/mmskeleton:/work --name mmskeleton mmdetection bash

2.安装mmskeleton

git clone https://github.com/open-mmlab/mmskeleton.git
python setup.py develop
python mmskl.py pose_demo

完成!

二、失败案例(来回折腾)

代码:https://github.com/yysijie/st-gcn

https://github.com/yysijie/st-gcn/blob/master/OLD_README.md

1.镜像:

我使用的是docker hub里面的镜像https://hub.docker.com/r/jaehwankimneo/st-gcn

2.运行发现 can not find openpose python api

然后就重新在容器里安了caffe、openpose,再装openpose时发现在build/python/openpose/路径下没有pyopenpose.cpython-35m-x86_64-linux-gnu.so文件。原因时当前镜像里的openpose太旧了,还没有python api 即便选择了BUILD_PYTHON也不会生成这个文件。所以从openpose官网(https://github.com/CMU-Perceptual-Computing-Lab/openpose)git下来新版本再进行cmake就产生了这个文件。

3. 运行报错,问题:运行显示显卡能力不够

Check failed: status == CUDNN_STATUS_SUCCESS (4 vs. 0)

我重装了cuda 、cudnn7、cudnn6、cudnn5.1都不对。。。。

还是有问题。。。。。。

慢慢折腾吧。。。。。。。。。

Published by

风君子

独自遨游何稽首 揭天掀地慰生平

发表回复

您的电子邮箱地址不会被公开。 必填项已用 * 标注