The issue does not reproduce consistently (about 50% flaky) and started appearing when we switched the OS image to one with a newer CUDA version (I don't know exactly what else could have changed, I ...