pytorch 版本>=1.8.0. 函数形态. torch.cuda.set_per_process_memory_fraction(0.5, 0). 参数1:fraction 限制的上限比例,如0.5 就是总GPU显存的一半,可以是0~1的 ...
確定! 回上一頁