OpenPose环境搭建(详细教程CPU/GPU)

作者:搬砖的石头2024.01.17 21:50浏览量:12

简介:在Windows 10上,使用Python 3.7,CUDA 11.6和VS2022,详细介绍如何搭建OpenPose环境。本教程将涵盖CPU和GPU环境的搭建过程。

千帆应用开发平台“智能体Pro”全新上线 限时免费体验

面向慢思考场景,支持低代码配置的方式创建“智能体Pro”应用

立即体验

首先,我们需要确保你已经安装了以下软件和工具:

  1. Windows 10操作系统
  2. Python 3.7
  3. Visual Studio 2022
  4. CUDA 11.6
    接下来,我们将按照以下步骤进行OpenPose环境的搭建:
    步骤一:安装依赖库
    在命令行中运行以下命令安装OpenPose所需的依赖库:
    pip install numpy opencv-python
    步骤二:获取OpenPose源代码
    你可以从OpenPose的GitHub仓库中获取最新的源代码。在命令行中运行以下命令:
    git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git
    步骤三:配置环境变量
    将OpenPose的路径添加到系统的环境变量中,以便在任何位置都可以访问OpenPose的相关文件。具体操作方法可以参考Windows的环境变量设置。
    步骤四:编译OpenPose(CPU模式)
    在命令行中进入OpenPose的源代码目录,然后运行以下命令进行编译:
    mkdir build
    cd build
    cmake .. -DCMAKE_BUILD_TYPE=RELEASE -DBUILD_PYTHON_BINDINGS=ON -DOPENCV_VERSION=4.5.1 -DCMAKE_INSTALL_PREFIX=../install -DCMAKE_PREFIX_PATH=C:\opencv\build
    make -j8
    make install
    以上命令将编译OpenPose的CPU版本,并生成Python绑定。编译完成后,将在指定的安装目录下生成OpenPose的相关文件。
    步骤五:编译OpenPose(GPU模式)
    如果你要使用GPU加速,你需要安装支持CUDA的NVIDIA显卡驱动和CUDA工具包。确保你的显卡驱动和CUDA版本兼容。然后,在命令行中进入OpenPose的源代码目录,运行以下命令进行编译:
    mkdir build_gpu
    cd build_gpu
    cmake .. -DBUILD_CPU_DEMOS=OFF -DBUILD_GPU_DEMOS=ON -DOPENCV_VERSION=4.5.1 -DCMAKE_INSTALL_PREFIX=../install -DCMAKE_PREFIX_PATH=C:\opencv\build;C:\cuda\include;C:\cuda\lib;C:\cuda\lib64 -DUSE_MSVC_RUNTIME_LIBRARY=ON -DOPENCV_CONFIG_PATH=C:\opencv\build\config\vc15\opencv451.cmake -DCMAKE_CXX_FLAGS=-DMSVC_RUNTIME_DLL -DMSVC_RUNTIME_TYPE=MT -DCMAKE_CUDA_FLAGS=”-gencode arch=compute_50,code=sm_50;” -DMSVC_RUNTIME_TYPE=MT -DUSE_PYTHON3=ON -DPYTHON3_EXECUTABLE=C:\Python37\python.exe -DPYTHON3_INCLUDE_DIR=C:\Python37\include -DPYTHON3_LIBRARIES=C:\Python37\libs\python37.lib -DPYTHON3_NUMPY_INCLUDE_DIRS=C:\Python37\Lib\site-packages\numpy\core\include -DBUILD_PYTHON_BINDINGS=ON -DOPENCV_VERSION=4.5.1 -DCMAKE_INSTALL_PREFIX=../install -DCMAKE_PREFIX_PATH=C:\opencv\build;C:\cuda\include;C:\cuda\lib;C:\cuda\lib64 -DUSE_MSVC_RUNTIME_LIBRARY=ON -DOPENCV_CONFIG_PATH=C:\opencv\build\config\vc15\opencv451.cmake -DCMAKE_CXX_FLAGS=-DMSVC_RUNTIME_DLL -DMSVC_RUNTIME_TYPE=MT -DCMAKE_CUDA_FLAGS=”-gencode arch=compute_50,code=sm_50;”
    make -j8
    make install
    以上命令将编译OpenPose的GPU版本,并生成Python绑定。编译完成后,将在指定的安装目录下生成OpenPose的相关文件。
    现在你已经成功搭建了OpenPose环境。你可以使用Python编写代码来调用Open
article bottom image

相关文章推荐

发表评论