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

yipeiwu_com4年前 (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设计】。

返回列表

本文采用OpenCV3和Python3 来实现静态图片的人脸识别,采用的是Haar文件级联。首先需要将OpenCV3源代码中找到data文件夹下面的haarcascades文件夹里...

本文实例为大家分享了python朋友圈九宫格图片的具体制作代码，供大家参考，具体内容如下将一张图片，切分成九宫格的样式：原图： # -*- coding: UTF-8 -*-...

本文实例讲述了python编程开发之日期操作。分享给大家供大家参考，具体如下：在python中对日期进行操作的库有： import datetime import time 对日期格式...

本文介绍了利用Python实现Windows下的鼠标键盘模拟的实例代码，分享给大家本来用按键精灵是可以实现我的需求，而且更简单，但既然学python ，就看一下呗。依赖： PyUs...

有时候，我们需要在字符串中加入相应的变量，以下提供了几种字符串加入变量的方法： 1、+ 连字符 name = 'zhangsan' print('my name is '+name...

[宜配屋]听图阁