Debayan
Databricks Employee
Databricks Employee

Hi, The issue can be with dependency package installation, which is CUDA here (most probably).

Also, could you please get the logs for full command output?

Also, please tag @Debayan​ with your next response which will notify me, Thank you!