Skip to content

Setup up Docker with GPUs

Erik Gärtner edited this page Nov 1, 2019 · 1 revision

Docker natively supports passing GPUs to container natively since Docker 19.03 through the --gpus-flag. Unfortunately the official Docker Python API doesn't (as of 2019-11-1) support this flag.

Therefore Hyperdock only supports GPU access by using a Docker version between 1.11 and 18.06, together with the nvidia2 runtime. To set this up on Ubuntu follow the steps below:

  1. Install apt package lists from Docker.
  2. Install the pinned Docker version 18.06.3~ce~3-0~ubuntu using apt install docker-ce=18.06.3~ce~3-0~ubuntu
  3. Install the nvidia2 runtime.
Clone this wiki locally