Hello @ppang !

Since you posted your question, the repository you shared has received an update, which includes the following warning:

"Using conda in DCS images is no longer supported starting Databricks Runtime 9.0. We highly recommend users to extend cuda-11.8 examples. We no longer support cuda-10.1 and cuda-11.0 compatibility with latest databricks runtime."

It's likely that the issue you encountered was related to a CUDA incompatibility.

Best regards,

Jéssica Santos