torch.cuda.memory.caching_allocator_delete

torch.cuda.memory.caching_allocator_delete(mem_ptr) [исходный код]

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

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

Параметры:

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

Примечание

Подробнее об управлении памятью GPU см. в разделе Управление памятью.