CUDAPluggableAllocator

class torch.cuda.CUDAPluggableAllocator(path_to_so_file, alloc_fn_name, free_fn_name)

從 so 檔案載入的 CUDA 記憶體分配器。