I trained my network on a gpu device and saved checkpoint by torch.save
Loading this checkpoint on my cpu device gives an error:
    raise AssertionError("Torch not compiled with CUDA enabled")
AssertionError: Torch not compiled with CUDA enabled```