I notice that when I run the same code on the same machine with a different version of PyTorch I get different VRAM consumption. With PyTorch 1.10.2 installed around Dec 2021 and Pytorch 1.8.1 ...
When running with the latest (1.9.0) wheel from here, as per the installation instructions, my project's Torch code crashes every time with an illegal instruction exception. The top 10 stack levels ...