网络编程
位置:首页>> 网络编程>> Python编程>> pytorch程序异常后删除占用的显存操作

pytorch程序异常后删除占用的显存操作

作者:upDiff  发布时间:2021-04-03 22:33:38 

标签:pytorch,程序异常,显存

1-删除模型变量


del model_define

2-清空CUDA cache


torch.cuda.empty_cache()

3-步骤2(异步)需要一定时间,设置时延


time.sleep(5)

完整代码如下:


del styler
torch.cuda.empty_cache()
time.sleep(5)

来源:https://blog.csdn.net/dlhlSC/article/details/86687789

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com