Hi @Revanth Pentyala​ 

Can you please try with DBR 7.3 ML cluster ?

It seems like cupti library was deprecated staring from 7.6 DBR.

https://docs.databricks.com/release-notes/runtime/7.6ml.html#deprecations

It seems the cupti version(9) which comes with ubuntu was not compatible with CUDA(11). The workaround would be to install the compatible cupti package(11) through init script.

However for now you can try with DBR 7.3 ML to see if it works there

Thanks

Mathan