CUDA 可插拔分配器 ¶

class torch.cuda.CUDAPluggableAllocator(动态库文件路径, 分配函数名, 释放函数名)[source][source] ¶

从 so 文件加载的 CUDA 内存分配器。