AssertionError: torch not compiled with Cuda enabled error occurs because of ...
assertionerror: no inf checks were recorded for this optimizer error occurs ...