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设计】。

返回列表

一、SQLalchemy简介 SQLAlchemy是一个开源的SQL工具包，基本Python编程语言的MIT许可证而发布的对象关系映射器。SQLAlchemy提供了“一个熟知的企业级全套...

本文实例讲述了python中栈的原理及实现方法。分享给大家供大家参考，具体如下：栈（stack），有些地方称为堆栈，是一种容器，可存入数据元素、访问元素、删除元素，它的特点在于只能允许...

本文实例讲述了Python 网络编程之TCP客户端/服务端功能。分享给大家供大家参考，具体如下： demo.py（TCP客户端）： import socket def main():...

前言 ROC（Receiver Operating Characteristic）曲线和AUC常被用来评价一个二值分类器（binary classifier）的优劣。这篇文章将先简单的介...

前言在学校太闲，就写了这个程序，可以在复平面绘制曲线，画圆什么的很轻松，f(z) = e^(1j * z)，螺旋线，函数图象等都可以。效果图说明此程序使用turtle绘图，可以...

宜配屋