Ubuntu命令行安装OpenPose详细指南

admin
Ubuntu命令行安装OpenPose需先安装依赖:CUDA、cuDNN、CMake、Boost等,确保版本兼容,接着克隆OpenPose仓库,执行git clone --recursive https://github.com/CMU-Perceptual-Computing-Lab/openpose.git,进入目录后,用mkdir build && cd build创建编译文件夹,运行cmake ..配置环境,再通过make -j$(nproc)编译,编译完成后,执行sudo make install安装,最后用./examples/demo/openpose.bin测试,注意CUDA架构需与硬件匹配,依赖库缺失会导致编译失败。

OpenPose是一款开源的实时2D人体姿态估计库,由卡内基梅隆大学开发,能够高效检测人体、手部、面部等关键点,广泛应用于计算机视觉、动作识别、人机交互等领域,本文将详细介绍在Ubuntu系统中通过命令行安装OpenPose的完整步骤,包括环境配置、依赖安装、源码编译及测试验证,帮助读者快速搭建OpenPose运行环境。

安装前准备

系统要求

  • 操作系统:Ubuntu 16.04/18.04/20.04(64位系统,推荐18.04 LTS,兼容性更好)
  • GPU:NVIDIA GPU(支持CUDA计算,显存建议至少4GB,如GTX 1060及以上)
  • 内存:至少8GB(编译过程需较多内存,建议16GB以上)

核心依赖安装

OpenPose基于Caffe深度学习框架开发,依赖CUDA、cuDNN、Boost、CMake等工具,以下是依赖安装的详细步骤:

(1)更新系统并安装基础工具

sudo apt update
sudo apt upgrade -y
sudo apt install -y build-essential cmake git wget unzip libprotobuf-dev libgoogle-glog-dev libboost-all-dev libhdf5-dev protobuf-compiler libatlas-base-dev

(2)安装NVIDIA驱动

OpenPose需GPU加速,需先安装NVIDIA驱动,可通过以下命令安装(根据GPU型号选择驱动版本,建议使用Ubuntu官方仓库的驱动或NVIDIA官网驱动):

Ubuntu命令行安装OpenPose详细指南

# 查看推荐的驱动版本
ubuntu-drivers devices
# 安装推荐驱动(以nvidia-driver-470为例)
sudo apt install nvidia-driver-470
# 重启系统使驱动生效
sudo reboot

重启后,通过nvidia-smi命令检查驱动是否安装成功(应显示CUDA版本和GPU信息)。

(3)安装CUDA Toolkit

OpenPose支持CUDA 9.0及以上版本,推荐CUDA 11.0(与Ubuntu 18.04/20.04兼容性更好),以下以CUDA 11.0为例:

  • NVIDIA CUDA官网下载对应Ubuntu版本的.run文件(如cuda_11.0.2_450.51.05_linux.run),或使用wget命令下载:

    wget https://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
  • 安装CUDA(需关闭图形界面,按Ctrl+Alt+F1进入命令行,登录后执行):

    sudo sh cuda_11.0.2_450.51.05_linux.run

    安装过程中,勾选"Accept"协议,并在"Driver Options"中取消"Install NVIDIA Accelerated Graphics Driver"(避免与之前安装的驱动冲突),仅安装CUDA Toolkit。

  • 配置CUDA环境变量:

    echo 'export PATH=/usr/local/cuda-11.0/bin:$PATH' >> ~/.bashrc
    echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
    source ~/.bashrc

    验证安装:nvcc --version应显示CUDA 11.0版本信息。

(4)安装cuDNN

cuDNN是NVIDIA的深度学习库加速库,需与CUDA版本匹配,以cuDNN 8.0.5(对应CUDA 11.0)为例:

  • NVIDIA cuDNN官网下载cuDNN(需注册NVIDIA开发者账号),选择"Download cuDNN v8.0.5 for CUDA 11.0",下载cudnn-11.0-linux-x64-v8.0.5.39.tar.gz

  • 解压并安装cuDNN:

    sudo tar -xzvf cudnn-11.0-linux-x64-v8.0.5.39.tar.gz -C /usr/local
    sudo chmod a+r /usr/local/cuda/include/cudnn*.h
    sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

    验证安装:cat /usr/local/cuda/include/cudnn_version.h | grep CUDNN_MAJOR -A 2应显示版本号8.0。

(5)安装Caffe(OpenPose依赖)

OpenPose需特定版本的Caffe,需从源码编译安装:

  • 安装Caffe依赖:
    sudo apt install -y libgflags-dev
文章版权声明:除非注明,否则均为xmsdn原创文章,转载或复制请以超链接形式并注明出处。

取消
微信二维码
微信二维码
支付宝二维码