Does Databricks Container Services (DCS) support for GPU containers with Databricks Runtime 11.3 LTS and higher?

ppang
New Contributor III

I have been trying to start a cluster using DCS with GPU containers (https://github.com/databricks/containers/tree/master/ubuntu/gpu), but was only successful with Databricks Runtime 10.4 LTS and lower.

With Databricks Runtime 11.3 LTS and higher, I got the error:

"Internal error message: Spark error: Driver down cause: driver state change"

Does DCS supports GPU containers for 11.3 LTS and higher?