1-删除模型变量
|
1
|
del model_define |
2-清空CUDA cache
|
1
|
torch.cuda.empty_cache() |
3-步骤2（异步）需要一定时间，设置时延
|
1
|
time.sleep(5) |
完整代码如下：
|
1
2
3
|
del styler torch.cuda.empty_cache() time.sleep(5) |
以上这篇pytorch程序异常后删除占用的显存操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持服务器之家。
原文链接：https://blog.csdn.net/dlhlSC/article/details/86687789
