Hi @Brian Gue Databricks container services for GPU clusters is now in public preview. You can build your docker image to use it on a GPU cluster.
https://docs.databricks.com/clusters/gpu.html#databricks-container-services-on-gpu-clusters
You can find the GPU-supported docker image in Github.
https://github.com/databricks/containers/tree/master/ubuntu/gpu
