torch.cuda.caching_allocator_alloc

torch.cuda.caching_allocator_alloc(size, device=None, stream=None)[源代码]

使用CUDA内存分配器进行内存分配。

为给定设备和流分配内存，以便与其它框架进行互操作。使用caching_allocator_delete()来释放分配的内存。

参数

• size (int) — 指定要分配的字节数。

• device (torch.device 或 int, 可选) – 选择的设备。如果为 None，则使用默认的 CUDA 设备。

• stream (torch.cuda.Stream 或 int, 可选) – 选择的流。如果为 None，则使用选定设备的默认流。

注意

关于GPU内存管理的更多信息，请参见内存管理。