最近在学习的过程中用到了mmpose,所以记录一下安装的过程。
我的系统环境是:ubuntu20.04
步骤也不算太多,也不算太难,其中还是有一些小坑需要大家注意一下。
大家在看我的记录的时候也可以参考一下mmpose官方给的安装说明,其中也讲的也比较详细。
1.创建一个新的conda环境(也可以用现有的),这样出问题了可以直接删掉方便重新搞。
#创建虚拟环境,环境名字是mmpose python版本是3.7(要求版本是3.6+) conda create -n mmpose python=3.7 #进入环境 conda activate mmpose
2.安装pytorch(要求pytorch版本1.3+),进入pytorch官网,选择合适的版本。
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html
安装结束可以试一试安装成功了没。
3 安装mmcv,要求是最新的
pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu11.1/torch1.8.1/index.html
这个命令需要根据自己的虚拟环境中cuda的版本和torch的版本去更改。
可以使用
conda list
去查看自己环境中的cuda的版本和torch的版本。以及确认自己是否成功安装mmcv
4.安装mmpose。
a.首先需要下载mmpose的代码文件。
#如果下载速度慢的话,可以自己去github下载 git clone https://github.com/open-mmlab/mmpose #下载完解压 cd mmpose
b.安装mmpose需要的一些包。
pip install -r requirements.txt
注意:在执行这个命令的过程中,安装某个包可能因为网络或者是某种原因会报错,大家就要根据所报的错误,进行手动安装。
大家可以在mmpose的文件里查看。
先打开requirements.txt
可以看到执行了在requirements文件中的四个txt文件,然后可以根据自己所报的错误,修改某个txt,然后自己手动去安装。例如在运行的时候,下载安装poseval时,特别慢。就可以把这个optional.txt中的poseval这一行删掉。
然后去到poseval的网址,自行下载解压,然后运行
cd poseval pip install -e .
然后poseval就安装成功了
最后一步(在mmpose的路径下)
pip install -v -e .
这样就大功告成了!!