diff --git a/Dockerfile b/Dockerfile index 3dd12dc..6149565 100644 --- a/Dockerfile +++ b/Dockerfile @@ -27,7 +27,7 @@ RUN echo "Installing dependencies..." && \ libcanberra-gtk-module \ libopencv-dev && \ python3 -m pip install \ - numpy \ + numpy==1.18.5 \ protobuf \ opencv-python diff --git a/README.md b/README.md index ccceca3..a9097eb 100644 --- a/README.md +++ b/README.md @@ -1,16 +1,36 @@ # openpose-docker -Dockerfile to build the excellent OpenPose software from CMU. -Ensure that you have `nvidia-docker` installed before you download this image. +Dockerfile to build the excellent [OpenPose](https://github.com/CMU-Perceptual-Computing-Lab/openpose) software from CMU. -To run the container, use the following commmand - +## Prerequisites + +Ensure that you have the latest version of Docker installed. For versions prior to 19.03, you may need to install `nvidia-docker`. Read more [here](https://github.com/NVIDIA/nvidia-docker). + +## Getting Started + +Pull the image from Docker Hub - -```bash -xhost + -docker run -it --net=host -e DISPLAY --runtime=nvidia ``` +$ docker pull exsidius/openpose +``` + +To run the container (name given: `openpose`), use the following commmands - + +``` +$ docker stop openpose && docker rm openpose +$ docker run -it \ + --gpus all \ + --name openpose \ + exsidius/openpose +``` + +Now, you should be inside the interactive docker container, i.e. +`root@:/openpose# ` + +_Note_: Remove the `--gpus all` flag if you want to run the container without GPUs. + +### Supports -Supports - 1. CUDA 10 -2. CUDnn 7 -3. Python 3 (will be 3.7 soon) +2. cuDNN 7 +3. Python 3.5.2 (will be 3.7 soon)