pytorch程序異常后刪除占用的顯存操作
1-刪除模型變量
del model_define
2-清空CUDA cache
torch.cuda.empty_cache()
3-步驟2(異步)需要一定時(shí)間,設(shè)置時(shí)延
time.sleep(5)
完整代碼如下:
del styler torch.cuda.empty_cache() time.sleep(5)
以上這篇pytorch程序異常后刪除占用的顯存操作就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Python?類中定義多個(gè)構(gòu)造器方法重載與泛方法
這篇文章主要為大家介紹了Python?類中定義多個(gè)構(gòu)造器方法重載與泛方法示例詳解,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2023-03-03

itchat和matplotlib的結(jié)合使用爬取微信信息的實(shí)例

Python 找出出現(xiàn)次數(shù)超過(guò)數(shù)組長(zhǎng)度一半的元素實(shí)例

python如何修改PYTHONPATH環(huán)境變量

如何使用python-opencv批量生成帶噪點(diǎn)噪線的數(shù)字驗(yàn)證碼

Python使用虛擬環(huán)境(安裝下載更新卸載)命令