Bin Bao

> LGTM! Do you plan to commit this eventually once all test pass? I have been thinking about doing this for a while to see if this helps reduce flakiness...

@pytorchbot rebase

@pytorchbot rebase

@pytorchbot rebase

https://hud.pytorch.org/pr/96904#12071246663 shows "RuntimeError: No CUDA GPUs are available"

@pytorchbot rebase

@pytorchbot rebase

I was hoping this can resolve the flaky accuracy failure on TIMM models, but `dla102` still fails once in a while (https://github.com/pytorch/pytorch/actions/runs/4450740075/jobs/7817003045).

@pytorchbot rebase

@malfet @seemethere , I was hoping this driver can help to fix the flaky test, but it doesn't. On the other hand, the new driver seems stable enough. Should I...