亚洲在线久爱草,狠狠天天香蕉网,天天搞日日干久草,伊人亚洲日本欧美

為了賬號安全,請及時綁定郵箱和手機立即綁定
已解決430363個問題,去搜搜看,總會有你想問的

如何通過刪除張量來釋放 gpu 內存?

如何通過刪除張量來釋放 gpu 內存?

aluckdog 2022-01-18 15:31:28
假設我創建了一個張量并將其放在 GPU 上,以后不需要它并想釋放分配給它的 GPU 內存;我該怎么做?import torcha=torch.randn(3,4).cuda() # nvidia-smi shows that some mem has been allocated.# do something# a does not exist and nvidia-smi shows that mem has been freed.我努力了:del adel a; torch.cuda.empty_cache()但它們都不起作用。
查看完整描述

1 回答

?
MYYA

TA貢獻1868條經驗 獲得超4個贊

運行del tensor會從 GPU 中釋放內存,但不會將其返回到設備,這就是為什么內存仍顯示為已使用的原因nvidia-smi。您可以創建一個新的張量,并重用該內存。

來源

https://discuss.pytorch.org/t/how-to-delete-pytorch-objects-correctly-from-memory/947 

https://discuss.pytorch.org/t/about-torch-cuda-empty-cache/34232


查看完整回答
反對 回復 2022-01-18
  • 1 回答
  • 0 關注
  • 364 瀏覽
慕課專欄
更多

添加回答

舉報

0/150
提交
取消
微信客服

購課補貼
聯系客服咨詢優惠詳情

幫助反饋 APP下載

慕課網APP
您的移動學習伙伴

公眾號

掃描二維碼
關注慕課網微信公眾號