Pytorch釋放顯存占用方式
更新時間：2020年01月13日 09:25:04 作者：冉冉升起的碼農(nóng)
今天小編就為大家分享一篇Pytorch釋放顯存占用方式，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
如果在python內(nèi)調(diào)用pytorch有可能顯存和GPU占用不會被自動釋放，此時需要加入如下代碼
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()
觀察由緩存分配器管理的內(nèi)存。
以上這篇Pytorch釋放顯存占用方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個參考，也希望大家多多支持腳本之家。
相關(guān)文章
這篇文章主要為大家詳細介紹了Python PyQt5中拖放效果的實現(xiàn)原理與實現(xiàn)代碼，文中的示例代碼講解詳細，感興趣的小伙伴可以了解一下2022-11-11
CentOS下Python3的安裝及創(chuàng)建虛擬環(huán)境的方法
這篇文章主要介紹了CentOS下Python3的安裝及創(chuàng)建虛擬環(huán)境的方法,本文給大家介紹的非常詳細，具有一定的參考借鑒價值，需要的朋友可以參考下2018-11-11
這篇文章主要為大家詳細介紹了python實現(xiàn)簡易五子棋游戲，文中示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2022-05-05
這篇文章主要介紹了Python發(fā)送email的3種方法,本文講解了使用登錄郵件服務(wù)器方法、調(diào)用sendmail命令、使用smtp服務(wù)來發(fā)送三種方法,需要的朋友可以參考下2015-04-04
簡單介紹Python中的try和finally和with方法
這篇文章主要介紹了Python中的try和finally和with方法,是Python學(xué)習(xí)當中的基礎(chǔ)知識,需要的朋友可以參考下2015-05-05
本文主要介紹了詳解pytest+Allure搭建方法以及生成報告常用操作，文中通過示例代碼介紹的非常詳細，具有一定的參考價值，感興趣的小伙伴們可以參考一下2021-09-09
這篇文章主要為大家詳細介紹了如何利用Python語言實現(xiàn)超快窗口截圖功能，可以自動獲取當前活動窗口并展示截圖，感興趣的可以了解一下2022-05-05
近來在學(xué)習(xí)用Python進行一些電腦運維的工作。所以下面這篇文章主要給大家介紹了關(guān)于利用python為運維人員寫一個監(jiān)控腳本的相關(guān)資料，文中通過示例代碼介紹的非常詳細，需要的朋友可以參考借鑒，下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧。2018-03-03
在Anaconda中如果沒有指定路徑,虛擬環(huán)境會默認安裝在anaconda所安裝的目錄下,這篇文章主要給大家介紹了關(guān)于anaconda虛擬環(huán)境默認路徑更改的相關(guān)資料,需要的朋友可以參考下2023-10-10
這篇文章主要介紹了Python爬蟲爬取博客實現(xiàn)可視化,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下2020-06-06