Skip to content

darts_utils.cuda.free_torch

Free the CUDA memory of pytorch.

Source code in darts-utils/src/darts_utils/cuda.py 
19
20
21
22
23
24
25
def free_torch():
    """Free the CUDA memory of pytorch."""
    import torch

    gc.collect()
    if torch.cuda.is_available():
        torch.cuda.empty_cache()