2022年最新AlphaPose环境配置（Linux+GPU）

Posted 2022-08-15 独角兽团队

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了2022年最新AlphaPose环境配置（Linux+GPU）相关的知识，希望对你有一定的参考价值。

1.AlphaPose简介

人体姿势识别AlphaPose是一款精准的多人姿态评估工具，是上海交通大学卢策吾团队开源的作品。

AlphaPose is an accurate multi-person pose estimator, which is the first open-source system that achieves 70+ mAP (75 mAP) on COCO dataset and 80+ mAP (82.1 mAP) on MPII dataset. To match poses that correspond to the same person across frames, we also provide an efficient online pose tracker called Pose Flow. It is the first open-source online pose tracker that achieves both 60+ mAP (66.5 mAP) and 50+ MOTA (58.3 MOTA) on PoseTrack Challenge dataset.

AlphaPose supports both Linux and Windows!

具体信息可前往GitHub查看GitHub - MVIG-SJTU/AlphaPose: Real-Time and Accurate Full-Body Multi-Person Pose Estimation&Tracking System

2.环境配置

1.我的环境

先介绍一下我的环境：Ubuntu 18.04.6 LTS (GNU/Linux 5.4.0-122-generic x86_64)

我已经预先安装了anaconda、git、cuda10.1，如果没有安装的话需要安装一下。anaconda和cuda的安装可参考其他文章，git的话直接pip install git即可

2.准备AlphaPose的环境

基本需求

1. 首先先创建一个用于AlphaPose的虚拟环境：conda create -n AlphaPose python=3.7

(Python的版本号可以修改，看个人需求)

如果想删除环境的话可以用：conda remove -n AlphaPose –all

2. 激活并进入虚拟环境：conda activate AlphaPose

3. 安装pytorch，去pytorch官网复制对应cuda版本的pytorch安装命令即可

Previous PyTorch Versions | PyTorch

因为我的cuda版本为10.1，所以选择了1.6.0版本的pytorch，命令为：

conda install pytorch==1.6.0 torchvision==0.7.0 cudatoolkit=10.1 -c pytorch

（如果下载速度慢，可以切换成国内镜像源下载，这里不赘述，自行查阅资料）

下载完成后，可以进入python终端，查看是否下载成功

python

import torch

print(torch.__version__) #注意是双下划线

4. 下载AlphaPose

用git命令直接clone到本地

git clone https://github.com/MVIG-SJTU/AlphaPose.git

如果想下载到指定目录下，可以采用git clone 源码地址本地路径的形式

（如果没有安装git，先用命令pip install git下载安装git）

进入文件夹中cd AlphaPose

3.配置AlphaPose

添加路径：

export PATH=/usr/local/cuda/bin/:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64/:$LD_LIBRARY_PATH

下载cpython库：python -m pip install cython
安装和更新：sudo apt-get install libyaml-dev
注意！！！
for users who want to use 1.5 < PyTorch < 1.11, please switch to the pytorch<1.11 branch by:
```
git checkout "pytorch<1.11"
```
for users who want to use PyTorch < 1.5, please switch to the pytorch<1.5 branch by:
```
 git checkout "pytorch<1.5"
```
以上完成后运行：python setup.py build develop

依据报错，执行相应操作。