Could you wipe the PyTorch installation inside the container via pip uninstall torch -y and install the latest binary?
Besides the OS change, PyTorch is also in another version (1.4.0 in Windows, while 20.01 uses a pre-1.4.0 build), so this might have been a known issue, was could have been fixed in 1.4.0.