Prabakar
Databricks Employee
Databricks Employee

Thanks for the update @Brian Gue​ . Hope, adding the rapids libraries to the cuda-image helps to resolve your issue.