'BatchNorm2d' object has no attribute 'track_running_stats'

Dear Ayush,
This might help.

Could I downgrade pytorch? or should I do something more after upgrading?

I used conda but I think the commands should be similar