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