欧美bbbwbbbw肥妇,免费乱码人妻系列日韩,一级黄片

pytorch程序異常后刪除占用的顯存操作

 更新時間:2020年01月13日 09:56:51   作者:upDiff  
今天小編就為大家分享一篇pytorch程序異常后刪除占用的顯存操作,具有很好的參考價值,希望對大家有所幫助。一起跟隨小編過來看看吧

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程序異常后刪除占用的顯存操作就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。

相關文章

最新評論