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

yipeiwu_com6年前 (2020-03-06)Python基础

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)

以上这篇pytorch程序异常后删除占用的显存操作就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持【听图阁-专注于Python设计】。

返回列表

python中getpass 模块的作用是输入密码不可见运行到这脚本不继续运行下去, 打开pycharm中的terminal 如上图显示，password中有输入密码，但不显示以...

1.算法：设有一组关键字｛ K 1 ， K 2 ，…， K n ｝；排序开始就认为 K 1 是一个有序序列；让 K 2 插入上述表长为 1 的有序序列，使之成为一个表长为 2 的有序序列...

python 中提供一种用于对函数固定属性的函数(与数学上的偏函数不一样) # 通常会返回10进制 int('12345') # print 12345 # 使用参数返回 8...

正则是处理字符串最常用的方法，我们编码中到处可见正则的身影。正则大同小异，python 中的正则跟其他语言相比略有差异： 1、替换字符串时，替换的字符串可以是一个函数 2、split...

本文实例讲述了python列表操作之extend和append的区别。分享给大家供大家参考。具体如下： li = ['a', 'b', 'c'] li.extend(['d', '...

宜配屋