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

yipeiwu_com5年前 (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是一个脚本语言，被解释器解释执行。它的发布方式： .py文件：对于开源项目或者源码没那么重要的，直接提供源码，需要使用者自行安装Python并且安装依赖的各种库。（Py...

如下所示： import http.client, urllib.parse import http.client, urllib.parse import random USER...

Python字符串字符串或串(String)是由数字、字母、下划线组成的一串字符。一般记为 : s="a1a2···an"(n>=0) 它是编程语言中表示文本的数据类...

本文实例讲述了Django框架用户注销功能实现方法。分享给大家供大家参考，具体如下： HttpResponse()里有个delete_cookie()方法专门用来删除cookie 我们到...

1.爬取豆瓣top250书籍 import requests import json import csv from bs4 import BeautifulSoup books =...

宜配屋