Hi,
On my machine with latest stable build of PyTorch (1.7.0) I can reproduce all examples you have mentioned. Maybe you should make sure you are using correct build of PyTorch. Unfortunately I cannot test docker edition.
Bests