Docker NVIDIA Support

docker-nvidia.md

1. Install NVIDIA container tools w/ this guide:

https://docs.nvidia.com/datacenter/cloud-native/container-toolkit/install-guide.html#docker

2. Add the following environment variables to docker-compose.yml

environment:
  - NVIDIA_VISIBLE_DEVICES=all
  - NVIDIA_DRIVER_CAPABILITIES=compute,video,utility

3. Set/Verify Default Docker Runtime

I can't remember if the nvidia-docker2 package did this or if I had to manually set it as the default runtime, so check \etc\docker\daemon.json to see if it looks like the following:

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