draugr.torch_utilities.system.device.torch_clean_up

draugr.torch_utilities.system.device.torch_clean_up() None[source]

Destroy cuda state by emptying cache and collecting IPC.

Consecutively calls torch.cuda.empty_cache() and torch.cuda.ipc_collect().