安装Open-MMlab 的 MMPose 和 MMCV

1. 基础环境

1.1 conda环境

anaconda的环境请自行准备,网上有一堆教程

conda create -n open-mmlab python=3.8 -y

1.2 pytorch 环境

因为我的CUDA是11.1 的所以下载这个
其他版本的CUDA可以参考官网的下载其他版本 ,参见pytorch官网:https://pytorch.org/get-started/locally/

pip install torch==1.9.0+cu111 torchvision==0.10.0+cu111 torchaudio===0.9.0 -f https://download.pytorch.org/whl/torch_stable.html

2. 安装MMCV

本来打算使用命令行就安装了,但是执行失败,搜索了才知道mmcv-full没有对应的windows编译好的版本,下面命令执行失败。

pip install mmcv-full -f https://download.openmmlab.com/mmcv/dist/cu111/torch1.9.0/index.html

2.1 MMCV 编译安装

因为mmcv没有windows版本的预编译,因此只能自己编译,(强烈推荐参照文档安装啊 )

2.2 安装记录

下面是在战神win10里面的最后的安装记录

2.2.1:设置环境变量

设置环境变量。添加 C:\Program Files (x86)\Microsoft Visual Studio\2019\Enterprise\VC\Tools\MSVC\14.29.30037\bin\Hostx86\x64PATH。则 cl.exe 可以在命令行中运行, 如下所示。
在这里插入图片描述

2.2.2 激活 anaconda的虚拟环境

进入anaconda的Prompt命令行
在这里插入图片描述
进入mmcv的源码下载目录
在这里插入图片描述

(base) PS D:\git\mmcv> conda activate open-mmlab

2.2.3 设置参数

(open-mmlib) PS D:\git\mmcv> $env:MMCV_WITH_OPS = 1
(open-mmlib) PS D:\git\mmcv> $env:MAX_JOBS = 8
(open-mmlib) PS D:\git\mmcv> $env:CUDA_HOME = "C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1"

其中TORCH_CUDA_ARCH_LIST这个表示算力,可以查询自己显卡的算力.
查询算力的命令在这个目录下:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.1\extras\demo_suite
deviceQuery.exe

在这里插入图片描述

强烈建议查询自己的算力,然后下一步的值填的精准才能正常编译!!!,不要填多个算力值,这样不仅编译很慢,而且有时候还不成功
填入自己算力↓

(open-mmlab) PS D:\git\mmcv> $env:TORCH_CUDA_ARCH_LIST="6.1"

2.2.4 安装依赖

(open-mmlab) PS D:\git\mmcv> pip3 install -r requirements.txt

2.2.5 开始编译

(open-mmlab) PS D:\git\mmcv> python setup.py build_ext  # 如果成功, cl 将会自动弹出来编译 flow_warp

安装pytest-runner↓

(open-mmlab) PS D:\git\mmcv> pip install pytest-runner

正式安装↓

(open-mmlab) PS D:\git\mmcv> python setup.py develop   # 安装

安装成功

2.2.5 查看安装

pip show mmcv-full

3. 安装MMPose

3.1 克隆 MMPose 库。

git clone https://github.com/open-mmlab/mmpose.git
cd mmpose

3.2 安装依赖包和 MMPose

pip install -r requirements.txt
pip install -v -e .  # or "python setup.py develop"

有个xtcocotools 没有,因此需要下载安装包:https://github.com/jin-s13/xtcocoapi
然后安装它:

pip install -r requirements.txt
python setup.py install
  • 3
    点赞
  • 18
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 3
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

炼丹狮

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值