中断脚本后清除GPU内存

时间:2020-06-01 12:54:37

标签: pytorch

我注意到,当我多次中断程序(在启用了cuda的Pytorch中编写)时,GPU内存不会被刷新。在这种情况下,如果脚本需要多次中断而没有完成执行,有什么方法可以解决?

0 个答案:

没有答案