I don't think this is specific to Databricks, but rather Tensorflow. See https://stackoverflow.com/questions/68614547/tensorflow-libdevice-not-found-why-is-it-not-found-in-t... for a possibly relevant solution.
I don't see evidence that this is related to libcupti
