Contents Menu Expand Light mode Dark mode Auto light/dark mode
DeLU
Logo
  • v0.0.21 ▼
  • Source code

API & Examples

  • delu
    • to
    • cat
    • iter_batches
  • delu.cuda
    • free_memory
  • delu.nn
    • Lambda
    • NLinear
    • named_sequential
  • delu.random
    • seed
    • get_state
    • set_state
    • preserve_state
  • delu.utils.data
    • Enumerate
    • IndexDataset
  • delu.tools
    • EarlyStopping
    • Timer

Deprecated API

  • [deprecated]
Back to top
Edit this page

delu.cuda#

An addition to torch.cuda.

free_memory

Free GPU memory = torch.cuda.synchronize + gc.collect + torch.cuda.empty_cache.

Next
free_memory
Previous
iter_batches
Copyright © 2023, Yura52
Made with Sphinx and @pradyunsg's Furo