Thanks. Yes, I can see the same behaviour for the latest stable release (PyTorch 1.9.0) if it is installed with Cuda 10.2. If it is installed with Cuda 11.1, the problem goes away as specified in the original post. Followed the conda specs here: https://pytorch.org/ 20