torch.cuda.caching_allocator_delete

torch.cuda.caching_allocator_delete(mem_ptr) [source]

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

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

Параметры

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

Примечание

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