Backward Loss function stops the program without any error or warning log on Windows about softras HOT 1 CLOSED

For anyone else who might have wandered on to this strange but probably rare issue, I solved it by simply installing pytorch version 10+. My current version 9.1 didn't work when any backward function was computed on it in CUDA ,worked fine for CPU. About the previous project in which I claimed it worked, I found that I was actually using torch version 10.1.
So here, for Windows, I had use torch 10.1 version. Pip installing the latest version (10.2), gets automatically downgraded to 9.1/9.2 which may have some compatibility issues with my current GPU drivers and thus resulted in this problem. So basically installing torch 10.1 works because pip doesn't downgrade 10.1 even though it downgrades 10.2 . Finally, you need to install Nvidia GPU Computing Toolkit too alongside this and make sure it matches the torch version , i.e 10.1 not 10.2

from softras.