nvidia-docker + docker-compose hello world

README.md

nvidia-docker with docker-compose

Setup

1. Install nvidia-docker-plugin
2. Reboot (for some reason)
3. On the folder with this docker-compose.yml file do

docker-compose up

You sould see your Nvidia card info.

Notes

Remember to map the devices. This is NOT specified on the documentation but its requiered for this to work.

Links

• https://github.com/NVIDIA/nvidia-docker/wiki/NVIDIA-driver
• NVIDIA/nvidia-docker#34
• NVIDIA/nvidia-docker#192

docker-compose.yml

version: '2'

volumes:
  nvidia_driver_367.57:
    external: true


services:
  test:
    image: nvidia/cuda
    volumes:
      - nvidia_driver_367.57:/usr/local/nvidia:ro
    devices:
      - /dev/nvidia0:/dev/nvidia0
      - /dev/nvidiactl:/dev/nvidiactl
      - /dev/nvidia-uvm:/dev/nvidia-uvm
    command: nvidia-smi