Prabakar
Databricks Employee
Databricks Employee

You need to choose the runtime for ML instead of the standard.

image