pytorch / 2 / generated / torch.mps.empty_cache.html

torch.mps.empty_cache

torch.mps.empty_cache() [source]

Releases all unoccupied cached memory currently held by the caching allocator so that those can be used in other GPU applications.

© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.mps.empty_cache.html