torch.cuda.caching_allocator_delete

torch.cuda.caching_allocator_delete(mem_ptr) [source]

Удаляет память, выделенную с помощью CUDA-аллокатора памяти.

Память, выделенная с помощью caching_allocator_alloc(), освобождается здесь. Связанное устройство и поток отслеживаются внутри аллокатора.

Параметры:

mem_ptr (int) – адрес памяти, который должен быть освобожден аллокатором.

Примечание

См. Управление памятью для получения более подробной информации об управлении памятью GPU.