-
openmim、mmcv、mmengine及mm其他库的安装顺序问题(安装顺序错了,会存在版本不对齐和相互链接不上的问题)
-
python版本问题—>建议版本在python3.6~3.9之间MMPose的py版本为3.8;MMDet的py版本为3.9
-
pytorch版本问题—>对于一些cuda版本较高的同学,不建议直接用pytorch2.0,安装与cuda对应的即可;如果是使用cpu的,则可以不考虑cuda版本
安装git for windows
点击download下载,保持默认逐步安装
安装conda(或者miniconda)
点击download 下载,保持默认逐步安装
基于显卡驱动安装cuda
下载安装
CUDA Toolkit 11.8 Downloads | NVIDIA Developer
选择Windows,x86_64,version10,exe(local)即可,点击download下载
查看是否安装成功
电脑搜索cmd,打开命令提示符。输入nvcc --version
使用conda创建虚拟环境
创建一个名为mycode的虚拟环境
conda create -n mycode python=3.9
激活环境
conda activate mycode
安装pytorch
pip3 install install torch==1.10.1+cu113 torchvision==0.11.2+cu113 torchaudio==0.10.1+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
安装mmcv
pip install -U openmim
mim install mmengine
mim install 'mmcv==2.0.0rc3'
mim install "mmdet>=3.0.0rc6"
安装其他依赖包
pip install opencv-python pillow matplotlib seaborn tqdm pycocotools -i https://pypi.tuna.tsinghua.edu.cn/simple
克隆代码
#如果下载速度慢的话,可以自己去github下载
git clone https://github.com/open-mmlab/mmpose
#下载完解压
cd mmpose
编译安装mmpose
mim install -e .
然后手动在D盘github文件夹下新建四个文件夹
D:/github/mmpose/checkpoint
D:/github/mmpose/outputs
D:/github/mmpose/data
D:/github/mmpose/data/test
下载测试图片和视频
切换回git bash,执行以下代码下载文件。如果git bash重开了,需要先进入mmpose目录下
cd d:/github/mmpose
# 多人图片,来源:https://www.pexels.com/zh-cn/photo/2168292/
wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220610-mmpose/images/multi-person.jpeg -O data/test/multi-person.jpeg
# 单人视频-唱跳篮球
wget https://zihao-openmmlab.obs.myhuaweicloud.com/20220610-mmpose/videos/cxk.mp4 -P data/test
# 妈妈和女儿跳舞,经微信压缩
wget https://zihao-openmmlab.obs.myhuaweicloud.com/20220610-mmpose/videos/mother_wx.mp4 -P data/test
# 两个女生跳舞视频,来源:https://mixkit.co/free-stock-video/two-girls-having-fun-in-a-retro-restaurant-42298/
wget https://zihao-openmmlab.obs.cn-east-3.myhuaweicloud.com/20220610-mmpose/videos/two-girls.mp4 -O data/test/two-girls.mp4
到此git bash可以关闭了
测试mmpose和mmcv
在conda环境下输入python,进入py环境
python
输入测试代码
import mmcv
import mmpose
print(mmcv.__version__)
print(mmpose.__version__)