the version of Pre_train model with torch

[ShaoCHis]

I use torch=1.9.1+cu111, load coc_tiny to validate and get an error "RuntimeError: CUDA error: out of memory".
I search something about it,there is a answer that maybe I use the different version torch with the training procession.
So I'd appreciate that if you can tell me what's the version of torch with the training procession.Thank you!!!!

[ShaoCHis]