Hey Frank, thank you for your attempt to reproduce the issue !



 KFrank:

Are you expecting this to be some kind of docker / configuration /
hardware issue, or do you think it’s a pure pytorch issue? I don’t see
it with my test. What versions of torch / torchvision are you using?
Can you reproduce your issue with a “raw” python / pytorch installation,
or does in only


Yes, I do reproduce the issue in a virutal env (ie. w/o docker). Using https://raw.githubusercontent.com/pytorch/pytorch/master/torch/utils/collect_env.py I get the environment specs below.
python collect_env.py
Collecting environment information...
PyTorch version: 1.10.0+cu102
Is debug build: False
CUDA used to build PyTorch: 10.2
ROCM used to build PyTorch: N/A

OS: Ubuntu 20.04.3 LTS (x86_64)
GCC version: (Ubuntu 9.3.0-17ubuntu1~20.04) 9.3.0
Clang version: Could not collect
CMake version: version 3.16.3
Libc version: glibc-2.31

Python version: 3.8.10 (default, Sep 28 2021, 16:10:42)  [GCC 9.3.0] (64-bit runtime)
Python platform: Linux-5.11.0-38-generic-x86_64-with-glibc2.29
Is CUDA available: False
CUDA runtime version: No CUDA
GPU models and configuration: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA
HIP runtime version: N/A
MIOpen runtime version: N/A

Versions of relevant libraries:
[pip3] mypy-extensions==0.4.3
[pip3] numpy==1.21.3
[pip3] torch==1.10.0
[pip3] torchvision==0.11.1
[conda] Could not collect