openpose-gpu

Turing optimized nvidia/cuda:10.1-cudnn7-runtime rebased openpose docker image

Install the nvidia docker hooks

distribution=$(. /etc/os-release;echo $ID$VERSION_ID)
curl -s -L https://nvidia.github.io/nvidia-container-runtime/$distribution/nvidia-container-runtime.repo | \
  sudo tee /etc/yum.repos.d/nvidia-container-runtime.repo
sudo yum install -y nvidia-container-runtime-hook nvidia-container-runtime
sudo service docker restart

For nvidia only docker host run the following

sudo mkdir -p /etc/systemd/system/docker.service.d
sudo tee /etc/systemd/system/docker.service.d/override.conf <<EOF
[Service]
ExecStart=
ExecStart=/usr/bin/dockerd --host=fd:// --add-runtime=nvidia=/usr/bin/nvidia-container-runtime
EOF
sudo systemctl daemon-reload
sudo systemctl restart docker

For general purpose docker host add the following to /etc/docker/daemon.json and restart the docker daemon

{
  "runtimes": {
      "nvidia": {
        "path": "/usr/bin/nvidia-container-runtime",
        "runtimeArgs": []
      }
  }
}

Run the container with following command

docker run -it --rm --runtime=nvidia cagdasbas/openpose-gpu:10.1-cudnn7-runtime bash