Check that a host system can use NVIDIA GPUs inside containers. Usage: `docker-compose run test-gpu`

docker-compose.yaml

version: "3"
services:
  test-gpu:
    image: tensorflow/tensorflow:latest-gpu
    command: python -c "import tensorflow as tf; assert tf.test.is_gpu_available()"
    deploy:
      resources:
        reservations:
          devices:
          - driver: nvidia
            capabilities: [gpu]