粗暴解決CUDA out of memory的問題
小渣渣復(fù)現(xiàn)大佬project發(fā)現(xiàn)GPU跑不動(dòng)，出現(xiàn)如下報(bào)錯(cuò)：
RuntimeError: CUDA out of memory.
看下來最簡單粗暴方法就是減少batch_size，慢是慢了不止一點(diǎn)點(diǎn)但至少跑得動(dòng)了！
補(bǔ)充：Pytorch GPU顯存充足卻顯示out of memory解決辦法
今天在測試一個(gè)pytorch代碼的時(shí)候顯示顯存不足，但是這個(gè)網(wǎng)絡(luò)框架明明很簡單，用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)文章
這篇文章主要介紹了Python部署web app的詳細(xì)教程,示例代碼基于Python2.x版本,需要的朋友可以參考下2015-04-04
Python3.4 splinter(模擬填寫表單)使用方法
今天小編就為大家分享一篇Python3.4 splinter(模擬填寫表單)使用方法，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2018-10-10
這篇文章主要介紹了如何在Python中引用其他模塊，在Python中，除了可以自定義模塊外，還可以引用其他模塊，主要包括使用標(biāo)準(zhǔn)庫和第三方模塊，下面文章分別詳細(xì)的介紹，需要的小伙伴可以參考一下2022-06-06
這篇文章主要介紹了Python使用Matplotlib繪制3D曲面圖,在Python中,使用Matplotlib庫繪制3D曲面圖可以通過mpl_toolkits.mplot3d模塊實(shí)現(xiàn),需要的朋友可以參考下2025-05-05
Python?Pandas高效處理Excel數(shù)據(jù)完整指南
在數(shù)據(jù)驅(qū)動(dòng)的時(shí)代,Excel仍是大量企業(yè)存儲(chǔ)核心數(shù)據(jù)的工具,Python的Pandas庫憑借其向量化計(jì)算、內(nèi)存優(yōu)化和豐富的數(shù)據(jù)處理接口,成為自動(dòng)化分析Excel數(shù)據(jù)的首選工具,下面我們就來看看具體應(yīng)用吧2025-05-05
這篇文章主要介紹了python高效的素?cái)?shù)判斷算法，研究算法的同學(xué)一定要看一下2021-04-04
這篇文章主要介紹了Pytorch BertModel的使用說明，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧2021-03-03
Python?ArcPy實(shí)現(xiàn)批量拼接長時(shí)間序列柵格圖像
這篇文章主要介紹了如何基于Python中ArcPy模塊，對(duì)大量不同時(shí)相的柵格遙感影像按照其成像時(shí)間依次執(zhí)行批量拼接的方法，感興趣的可以了解一下2023-03-03
Python實(shí)例一個(gè)類背后發(fā)生了什么
Python實(shí)例一個(gè)類背后發(fā)生了什么，本文為大家一一列出，感興趣的朋友可以參考一下2016-02-02
python中sklearn庫predict及python sklearn linearmodel(
Scikit-learn(sklearn)是機(jī)器學(xué)習(xí)中常用的第三方模塊,對(duì)常用的機(jī)器學(xué)習(xí)方法進(jìn)行了封裝,包括回歸(Regression)、降維(Dimensionality Reduction)、分類(Classfication)、聚類(Clustering)等方法,今天小編給大家分享python中sklearn庫predict的問題,感興趣的朋友一起看看吧2024-02-02