No, unfortunately the code fails in such a way that there is no stack trace, it simply stalls. The program will not even respond to ctrl+c. The best I could do regarding a stack trace was by using the faulthandler library, which returns the trace you see in my post.
With your hardware question - RAM, CPU, Disk Usage, and GPU memory usage all seem fine. I would be inclined to think it’s not a hardware issue since the issue happens both on AI-platform and a custom VM. Perhaps there’s an incompatibility with the P100 in particular? I don’t know how likely that is.
Happy to try to some things out if you have any suggestions.