Pytorch GPU顯存充足卻顯示out of memory的解決方式
更新時(shí)間：2020年01月13日 10:14:08 作者：imaginist233
今天小編就為大家分享一篇Pytorch GPU顯存充足卻顯示out of memory的解決方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧
今天在測(cè)試一個(gè)pytorch代碼的時(shí)候顯示顯存不足，但是這個(gè)網(wǎng)絡(luò)框架明明很簡(jiǎn)單，用CPU跑起來(lái)都沒(méi)有問(wèn)題，GPU卻一直提示out of memory.
在網(wǎng)上找了很多方法都行不通，最后我想也許是pytorch版本的問(wèn)題，原來(lái)我的pytorch版本是0.4.1，于是我就把這個(gè)版本卸載，然后安裝了pytorch1.1.0，程序就可以神奇的運(yùn)行了，不會(huì)再有OOM的提示了。雖然具體原因還不知道為何，這里還是先mark一下，具體過(guò)程如下：
卸載舊版本pytorch：
conda uninstall pytorch
安裝pytorch1.1.0，按照官網(wǎng)上的辦法，我的CUDA版本是9.0：
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch
大功告成!
以上這篇Pytorch GPU顯存充足卻顯示out of memory的解決方式就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
lxml是python的一個(gè)解析庫(kù),支持HTML和XML的解析,支持XPath解析方式，下面這篇文章主要給大家介紹了關(guān)于python3解析庫(kù)lxml的安裝與使用的相關(guān)資料，文中通過(guò)示例代碼介紹的非常詳細(xì)，需要的朋友可以參考下2018-06-06
pandas groupby分組對(duì)象的組內(nèi)排序解決方案
這篇文章主要介紹了pandas groupby分組對(duì)象的組內(nèi)排序解決方案，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧2021-03-03
如何使用Django Admin管理后臺(tái)導(dǎo)入CSV
這篇文章主要介紹了如何使用Django Admin管理后臺(tái)導(dǎo)入CSV，幫助大家更好的理解和使用django框架，感興趣的朋友可以了解下2020-11-11
PyQt子線(xiàn)程處理業(yè)務(wù)事件的問(wèn)題解決
在PyQt中,主線(xiàn)程通常是指GUI主循環(huán)所在的線(xiàn)程,而子線(xiàn)程則是執(zhí)行實(shí)際工作的線(xiàn)程,本文主要介紹了PyQt子線(xiàn)程處理業(yè)務(wù)事件的問(wèn)題解決,具有一定的參考價(jià)值,感興趣的可以了解一下2024-02-02
Python數(shù)據(jù)分析:pandas中Dataframe的groupby與索引用法
這篇文章主要介紹了pandas中Dataframe的groupby與索引用法,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-02-02
這篇文章主要介紹了Python 描述符descriptor使用原理解析,文中通過(guò)示例代碼介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友可以參考下2020-03-03
這篇文章主要為大家詳細(xì)介紹了python讀取mysql數(shù)據(jù)繪制條形圖，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價(jià)值，感興趣的小伙伴們可以參考一下2020-03-03
這篇文章主要介紹了python使用calendar輸出指定年份全年日歷的方法,涉及Python使用calendar模塊操作日期的技巧,非常具有實(shí)用價(jià)值,需要的朋友可以參考下2015-04-04
Python線(xiàn)程之如何解決共享變量問(wèn)題
這篇文章主要介紹了Python線(xiàn)程之如何解決共享變量問(wèn)題，掐滅問(wèn)我們學(xué)習(xí)了銀行轉(zhuǎn)賬的這個(gè)場(chǎng)景，本文解決上次多個(gè)線(xiàn)程的操作都更改了amount變量導(dǎo)致運(yùn)行結(jié)果不對(duì)的問(wèn)題,需要的朋友可以參考一下2022-02-02
Python SQLAlchemy入門(mén)教程(基本用法)
這篇文章主要介紹了Python SQLAlchemy入門(mén)教程,本文通過(guò)實(shí)例主要給大家講解了python SQLAlchemy基本用法，需要的朋友可以參考下2019-11-11