a = torch.cuda.memory_reserved(0)
print(a)
pytorch查看当前显存分配
最新推荐文章于 2024-04-11 17:48:42 发布
a = torch.cuda.memory_reserved(0)
print(a)