And I meet this:
ImportError: cannot import name ‘size_based_auto_wrap_policy’ from ‘torch.distributed.fsdp.wrap’
Could anyone provide some suggestion? Thank you!!
In my win10, I have pytorch 1.12.1 py3.9_cuda11.6_cudnn8_0 pytorch
In my linux server, I have torch 1.12.0a0+bd13bc6 pypi_0 pypi
my win10 can find the size_based… but server can not!! Why?
Sorry, I am still a learner…I am not sure what is ‘server build’? Is that someting I can change? Or a server hardware that I can not change? (Previously I thought I can update pytorch to solve this…)
Thank you!!!