gpu pytorch 显存

1、torch.cuda.empty_cache() 释放GPU显存空间

torch.cuda.empty_cache()

2、torch.cuda.memory_summary() 查看显存信息

print(torch.cuda.memory_summary())

经测试，这种方法能释放掉一部分内存，效果不明显。最好的办法就是杀掉进程，或关闭kernel。