Shortcuts

torch.cuda.change_current_allocator

torch.cuda.change_current_allocator(allocator)[source]

Changes the currently used memory allocator to be the one provided. If the current allocator has already been used/initialized, this function will error.

Parameters:

allocator (torch.cuda.memory._CUDAAllocator) – allocator to be set as the active one.

Note

See Memory management for details on creating and using a custom allocator

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources