而且PyTorch是有缓存区的设置的,意思就是一个Tensor就算被释放了,进程也不 ... print(torch.cuda.memory_allocated() - temp) # 第一次增加4194304.
確定! 回上一頁