Rate this Page
★ ★ ★ ★ ★

torch.cuda.memory.caching_allocator_disabled#

torch.cuda.memory.caching_allocator_disabled()[source]#

Context manager that temporarily disables the CUDA caching allocator.

On this page
Show Source
PyTorch Libraries