pytorch_lightning.utilities.memory module¶

pytorch_lightning.utilities.memory.garbage_collection_cuda()[source]¶: Garbage collection Torch (CUDA) memory.

pytorch_lightning.utilities.memory.is_cuda_out_of_memory(exception)[source]¶

pytorch_lightning.utilities.memory.is_cudnn_snafu(exception)[source]¶

pytorch_lightning.utilities.memory.is_oom_error(exception)[source]¶

pytorch_lightning.utilities.memory.is_out_of_cpu_memory(exception)[source]¶

pytorch_lightning.utilities.memory.recursive_detach(in_dict)[source]¶

Detach all tensors in in_dict.

May operate recursively if some of the values in in_dict are dictionaries which contain instances of torch.Tensor. Other types in in_dict are not affected by this utility function.

Parameters: in_dict¶ (dict) –
Returns
Return type: out_dict