Adding GPU capability to a Docker workpool (equivalent of "--gpus all" flag)

[AmagicalFishy]

Just adding this here in case it helps anyone in the future:

After creating a Docker workpool, edit the Base Job Template JSON (you can do this in the UI via editing the workpool and clicking Advanced). Add to the "job_configuration" dictionary the following key-dict pair:

"container_create_kwargs": {
      "device_requests": [
        {
          "count": -1,
          "capabilities": [
            [
              "gpu"
            ]
          ]
        }
      ]
    }

This is the equivalent of adding --gpus all to e.g. docker run.

I suggest something like this be put in the documentation, I suspect it'll be an increasingly popular use-case (big thank you to @kevingrismore for adding this description ❤️ )

[zzstoatzz]

thanks so much for creating this discussion @AmagicalFishy ! we'll get this added to the docs as a how-to