Pytorch釋放顯存占用方式-創(chuàng)新互聯

如果在python內調用pytorch有可能顯存和GPU占用不會被自動釋放，此時需要加入如下代碼

網站建設哪家好，找創(chuàng)新互聯建站！專注于網頁設計、網站建設、微信開發(fā)、重慶小程序開發(fā)、集團企業(yè)網站建設等服務項目。為回饋新老客戶創(chuàng)新互聯還提供了青州免費建站歡迎大家使用！

torch.cuda.empty_cache()

我們來看一下官方文檔的說明

Releases all unoccupied cached memory currently held by the caching allocator so that those can be used in other GPU application and visible in nvidia-smi.

Note

empty_cache() doesn't increase the amount of GPU memory available for PyTorch. See Memory management for more details about GPU memory management.

此外還可以使用

memory_allocated()和max_memory_allocated()

觀察顯存占用，并使用

memory_cached()和 max_memory_cached()

觀察由緩存分配器管理的內存。

以上這篇Pytorch釋放顯存占用方式就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持創(chuàng)新互聯成都網站設計公司。

另外有需要云服務器可以了解下創(chuàng)新互聯scvps.cn，海內外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業(yè)上云的綜合解決方案，具有“安全穩(wěn)定、簡單易用、服務可用性高、性價比高”等特點與優(yōu)勢，專為企業(yè)上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。

網站題目：Pytorch釋放顯存占用方式-創(chuàng)新互聯
當前鏈接：http://bm7419.com/article0/diooio.html

成都網站建設公司_創(chuàng)新互聯，為您提供網站建設、微信小程序、虛擬主機、建站公司、網站制作、Google

廣告

聲明：本網站發(fā)布的內容（圖片、視頻和文字）以用戶投稿、用戶轉載內容為主，如果涉及侵權請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網站立場，如需處理請聯系客服。電話：028-86922220；郵箱：631063699@qq.com。內容未經允許不得轉載，或轉載時需注明來源： 創(chuàng)新互聯