This commit is contained in:
louiscklaw
2025-01-31 22:09:43 +08:00
parent abf2c0c67b
commit 97df42e0d5
5 changed files with 75 additions and 0 deletions

Binary file not shown.

View File

View File

@@ -0,0 +1,36 @@
# https://hub.docker.com/r/cwaffles/openpose
FROM ubuntu:20.04
#get deps
RUN apt-get update
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends python3-dev python3-pip python3-setuptools
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends git wget
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends g++
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends make
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libprotobuf-dev
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends protobuf-compiler
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libopencv-dev
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libgoogle-glog-dev
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libboost-all-dev
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libcaffe-cuda-dev
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libhdf5-dev
RUN DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends libatlas-base-dev
#for python api
RUN pip3 install --upgrade pip
RUN pip3 install numpy opencv-python
#replace cmake as old version has CUDA variable bugs
RUN wget https://github.com/Kitware/CMake/releases/download/v3.16.0/cmake-3.16.0-Linux-x86_64.tar.gz && \
tar xzf cmake-3.16.0-Linux-x86_64.tar.gz -C /opt && \
rm cmake-3.16.0-Linux-x86_64.tar.gz
ENV PATH="/opt/cmake-3.16.0-Linux-x86_64/bin:${PATH}"
#get openpose
WORKDIR /openpose
RUN git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git .
#build it
WORKDIR /openpose/build
RUN cmake -DBUILD_PYTHON=ON .. && make -j `nproc`
WORKDIR /openpose

View File

@@ -0,0 +1,27 @@
# https://hub.docker.com/r/cwaffles/openpose
FROM nvidia/cuda:11.4.0-cudnn8-devel-ubuntu18.04
#get deps
RUN apt-get update && \
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
python3-dev python3-pip python3-setuptools git g++ wget make libprotobuf-dev protobuf-compiler libopencv-dev \
libgoogle-glog-dev libboost-all-dev libcaffe-cuda-dev libhdf5-dev libatlas-base-dev
#for python api
RUN pip3 install --upgrade pip
RUN pip3 install numpy opencv-python
#replace cmake as old version has CUDA variable bugs
RUN wget https://github.com/Kitware/CMake/releases/download/v3.16.0/cmake-3.16.0-Linux-x86_64.tar.gz && \
tar xzf cmake-3.16.0-Linux-x86_64.tar.gz -C /opt && \
rm cmake-3.16.0-Linux-x86_64.tar.gz
ENV PATH="/opt/cmake-3.16.0-Linux-x86_64/bin:${PATH}"
#get openpose
WORKDIR /openpose
RUN git clone https://github.com/CMU-Perceptual-Computing-Lab/openpose.git .
#build it
WORKDIR /openpose/build
RUN cmake -DBUILD_PYTHON=ON .. && make -j `nproc`
WORKDIR /openpose

View File

@@ -0,0 +1,12 @@
# openpose-docker
A docker build file for CMU openpose with Python API support
https://hub.docker.com/r/cwaffles/openpose
### Requirements
- Nvidia Docker runtime: https://github.com/NVIDIA/nvidia-docker#quickstart
- CUDA 10.0 or higher on your host, check with `nvidia-smi`
### Example
`docker run -it --rm --gpus all -e NVIDIA_VISIBLE_DEVICES=0 cwaffles/openpose`
The Openpose repo is in `/openpose`