Doesn’t seem to work. In fact, turning off benchmark mode and going back to just using a smaller input, I tried to iterate on several gradient descents, and eventually I get the same kind of sudden crash. The system’s log file offers no information about it. The computer doesn’t even reboot afterwards unless I physically disconnect it before trying again. torch.cuda.empty_cache() made no difference, nor did running the code from the interpreter.