torch.cuda.memory.caching_allocator_enable

torch.cuda.memory.caching_allocator_enable(value=True)

啟用或停用 CUDA 記憶體分配器。預設為開啟。