粗暴解決CUDA out of memory的問題
小渣渣復(fù)現(xiàn)大佬project發(fā)現(xiàn)GPU跑不動(dòng)，出現(xiàn)如下報(bào)錯(cuò)：
RuntimeError: CUDA out of memory.
看下來最簡(jiǎn)單粗暴方法就是減少batch_size，慢是慢了不止一點(diǎn)點(diǎn)但至少跑得動(dòng)了！
補(bǔ)充：Pytorch GPU顯存充足卻顯示out of memory解決辦法
今天在測(cè)試一個(gè)pytorch代碼的時(shí)候顯示顯存不足，但是這個(gè)網(wǎng)絡(luò)框架明明很簡(jiǎn)單，用CPU跑起來都沒有問題，GPU卻一直提示out of memory.
在網(wǎng)上找了很多方法都行不通，最后我想也許是pytorch版本的問題，原來我的pytorch版本是0.4.1，于是我就把這個(gè)版本卸載，然后安裝了pytorch1.1.0，程序就可以神奇的運(yùn)行了，不會(huì)再有OOM的提示了。雖然具體原因還不知道為何，這里還是先mark一下，
具體過程如下：
卸載舊版本pytorch：
conda uninstall pytorch
安裝pytorch1.1.0，按照官網(wǎng)上的辦法，我的CUDA版本是9.0：
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
大功告成！
以上為個(gè)人經(jīng)驗(yàn)，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
VScode查看python f.write()的文件亂碼問題及解決方法
這篇文章主要介紹了VScode查看python f.write()的文件亂碼問題及解決方法,本文通過圖文并茂的形式給大家分享解決方法，需要的朋友可以參考下2023-02-02
這篇文章主要介紹了在Python的Django框架中包裝視圖函數(shù)的方法,即requires_login的相關(guān)方法,需要的朋友可以參考下2015-07-07
python如何爬取網(wǎng)站數(shù)據(jù)并進(jìn)行數(shù)據(jù)可視化
這篇文章主要介紹了python爬取拉勾網(wǎng)數(shù)據(jù)并進(jìn)行數(shù)據(jù)可視化,爬取拉勾網(wǎng)關(guān)于python職位相關(guān)的數(shù)據(jù)信息，并將爬取的數(shù)據(jù)已csv各式存入文件，然后對(duì)csv文件相關(guān)字段的數(shù)據(jù)進(jìn)行清洗，并對(duì)數(shù)據(jù)可視化展示，包括柱狀圖展示、直方圖展示,需要的朋友可以參考下2019-07-07
詳解Python字符串對(duì)象的實(shí)現(xiàn)
本文介紹了 python 內(nèi)部是如何管理字符串對(duì)象，以及字符串查找操作是如何實(shí)現(xiàn)的，感興趣的小伙伴們可以參考一下2015-12-12
這篇文章主要介紹了Python編程實(shí)現(xiàn)蟻群算法詳解，涉及螞蟻算法的簡(jiǎn)介，主要原理及公式，以及Python中的實(shí)現(xiàn)代碼，具有一定參考價(jià)值，需要的朋友可以了解下2021-10-10
Python面向?qū)ο蟪绦蛟O(shè)計(jì)類變量與成員變量、類方法與成員方法用法分析
這篇文章主要介紹了Python面向?qū)ο蟪绦蛟O(shè)計(jì)類變量與成員變量、類方法與成員方法用法,結(jié)合實(shí)例形式較為詳細(xì)的分析了類變量與成員變量、類方法與成員方法、類方法與靜態(tài)方法等概念、原理及使用技巧,需要的朋友可以參考下2019-04-04
今天給大家?guī)淼氖顷P(guān)于Python的相關(guān)知識(shí),文章圍繞著如何用Python實(shí)現(xiàn)感知器算法展開,文中有非常詳細(xì)的介紹及代碼示例,需要的朋友可以參考下2021-06-06
Python按行讀取文件的簡(jiǎn)單實(shí)現(xiàn)方法
下面小編就為大家?guī)硪黄狿ython按行讀取文件的簡(jiǎn)單實(shí)現(xiàn)方法。小編覺得挺不錯(cuò)的，現(xiàn)在就分享給大家，也給大家做個(gè)參考。一起跟隨小編過來看看吧2016-06-06
這篇文章主要介紹了Django如何實(shí)現(xiàn)防止XSS攻擊,文中通過示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-10-10
在?Python?中,斷點(diǎn)是一種在代碼中設(shè)置的標(biāo)記,用于在程序執(zhí)行過程中停止或中斷程序的執(zhí)行,本文將詳細(xì)介紹?Python?中的斷點(diǎn)類型,感興趣的可以了解下2024-02-02