OpenPose项目安装指南:系统环境准备详解
openpose 项目地址: https://gitcode.com/gh_mirrors/op/openpose
前言
OpenPose作为一款优秀的人体姿态识别开源库,其安装过程需要做好充分的环境准备工作。本文将详细介绍在不同操作系统下安装OpenPose前需要满足的系统要求,帮助开发者顺利完成环境配置。
通用准备事项
在开始安装前,有几个关键点需要注意:
-
CUDA版本兼容性:不同型号的NVIDIA显卡对CUDA版本有不同要求。例如:
- 30系列显卡至少需要CUDA 11
- 20系列、V100、Volta或Turing架构显卡至少需要CUDA 10
- 更早的显卡可能需要CUDA 8
-
模型文件下载:OpenPose依赖多个预训练模型,CMake会自动下载这些模型。但在某些网络环境下(如公司内网或有防火墙限制),可能需要手动下载并放置到指定目录:
- 身体姿态模型:
models/pose/body_25/
- COCO模型:
models/pose/coco/
- MPI模型:
models/pose/mpi/
- 面部模型:
models/face/
- 手部模型:
models/hand/
- 身体姿态模型:
Ubuntu系统准备
1. 环境检查
重要提示:如果系统安装了Anaconda,需要特别注意:
- Anaconda自带的Protobuf版本可能与Caffe不兼容
- 解决方案:
- 卸载Anaconda并通过apt-get安装protobuf
- 或者执行
conda deactivate
命令退出conda环境(可能需要执行两次)
2. CMake GUI安装
不同Ubuntu版本安装方式不同:
- Ubuntu 20:直接运行
sudo apt-get install cmake-qt-gui
- Ubuntu 18:
- 需要从源码编译安装CMake GUI(3.12及以上版本)
- 具体步骤包括卸载现有版本、安装依赖库、下载源码编译等
- Ubuntu 14/16:可直接通过apt-get安装
3. NVIDIA GPU版本准备
推荐配置:
- Ubuntu 20:CUDA 11.7.1 + cuDNN 8.5.0
- Ubuntu 18:CUDA 10.1 + cuDNN 7.5.1
- Ubuntu 14/16:CUDA 8 + cuDNN 5.1
安装步骤:
- 升级NVIDIA驱动至最新版本
- 安装CUDA:
- 可通过脚本自动安装:
sudo bash ./scripts/ubuntu/install_cuda.sh
- 或手动从官网下载安装
- 可通过脚本自动安装:
- 安装cuDNN:
- 下载对应版本后,将文件复制到CUDA安装目录
4. AMD GPU版本准备
仅在没有NVIDIA显卡时考虑:
- Ubuntu 20/18:官方未测试,建议使用CPU版本
- Ubuntu 14/16:
- 安装ROCM驱动
- 安装ViennaCL库:
sudo apt-get install libviennacl-dev
5. 依赖库安装
需要安装以下关键库:
- OpenCV:
sudo apt-get install libopencv-dev
- Caffe依赖:
sudo bash ./scripts/ubuntu/install_deps.sh
- 其他必要库:
sudo apt install protobuf-compiler libgoogle-glog-dev sudo apt install libboost-all-dev libhdf5-dev libatlas-base-dev
6. Python API准备(可选)
如需使用Python接口:
# Python 3
sudo apt-get install python3-dev
sudo pip3 install numpy opencv-python
# Python 2
sudo apt-get install python-dev
sudo pip install numpy opencv-python
MacOS系统准备
-
安装Homebrew(如未安装):
bash scripts/osx/install_brew.sh
-
安装CMake GUI:
brew install --cask cmake
-
安装依赖库:
bash scripts/osx/install_deps.sh
Windows系统准备
1. 基础工具安装
- CMake GUI:从官网下载Windows安装包
- Visual Studio:
- 推荐使用VS2019 Enterprise
- 安装时必须勾选所有C++相关组件
2. NVIDIA GPU版本准备
推荐配置:
- VS2019:CUDA 11.1.1 + cuDNN 8.1.0
- VS2017:CUDA 10.1 + cuDNN 7.5.1
- VS2015:CUDA 8 + cuDNN 5.1
安装注意事项:
- 必须先安装VS再安装CUDA
- 如遇CUDA相关错误,建议按顺序:重装VS → 重启 → 重装CUDA
3. AMD GPU版本准备
- 从AMD官网下载最新驱动
- ViennaCL库已包含在OpenPose中,无需额外安装
4. 依赖库准备
CMake会自动下载所需DLL,也可手动下载并放置到3rdparty/windows/
目录下:
- Caffe(根据需求选择CUDA/CPU/OpenCL版本)
- Caffe依赖库
- OpenCV 4.2.0
5. Python API准备(可选)
- 安装Python 3.x
- 安装必要包:
pip install numpy opencv-python
总结
OpenPose的安装环境准备因操作系统和硬件配置而异。本文详细介绍了Ubuntu、MacOS和Windows三大平台下的准备工作,特别是针对不同显卡和CUDA版本的配置建议。正确完成这些准备工作将为后续的OpenPose编译和运行打下坚实基础。
创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考