Windows下RTMpose环境部署

最新推荐文章于 2024-07-19 14:33:28 发布

广州记忆

最新推荐文章于 2024-07-19 14:33:28 发布

阅读量184

点赞数

文章标签： windows pytorch 人工智能

本文链接：https://blog.csdn.net/vitor_lxy/article/details/134887734

版权

本文详细介绍了如何下载MMpose源码，创建和验证虚拟环境，安装MMCV及其依赖，创建文件夹并下载模型，以及进行初步的测试。

摘要由CSDN通过智能技术生成

1. 环境配置

1.1 MMpose源码下载

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

1.2 MMpose虚拟环境创建

1.2.1 创建虚拟环境

conda create --name mmpose python=3.10
conda activate mmpose
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu116

1.2.2 虚拟环境验证

cd mmpose
python
import torch
torch.__version__
torch.cuda.is_available()

如果输出为True 表示配置成功

1.3 安装MMCV

1.3.1 安装openmim

pip install –U openmim

1.3.2 安装mmengine、mmcv和mmdet

mim install mmengine
mim install mmcv
mim install mmdet

1.3.3 安装其他依赖

mim install –e .

1.3.3 验证是否安装成功

执行以下指令，能够正常返回相关的版本号表示安装成功

python
import mmcv
from mmcv.ops import get_compiling_cuda_version, get_compiler_version
mmcv.__version__
get_compiling_cuda_version()
get_compiler_version()

MMpose版本确认

python
import mmpose
mmpose.__version__

2. 模型下载

2.1 创建相关文件夹

python
import os
os.mkdir('checkpoint')
os.mkidr('outputs')
os.mkidr('data')
os.mkidr('data/test')

2.2 下载模型

3. 测试