torch.cuda.set_per_process_memory_fraction¶
- torch.cuda.set_per_process_memory_fraction(fraction, device=None)[源代码]¶
设置进程的内存占用比例。
该分数用于限制CUDA设备上的缓存分配器分配的内存。 允许的值等于总可见内存乘以该分数。 如果在进程中尝试分配超过允许值的内存，将在分配器中引发内存不足错误。
- Parameters
fraction (float) – 范围：0~1。允许的内存等于总内存 * fraction。
设备 (torch.device 或 整数, 可选) – 选择的设备。如果它是
None，则使用默认的CUDA设备。
注意
一般来说，总可用空闲内存小于总容量。