I am using the library and would like to write tests that use GPU. I looked around and didn't see code, nor issue, for that. Is there a way to use GPUs in the current state of implementation?

Context:
https://docs.docker.com/engine/containers/resource_constraints/#expose-gpus-for-use