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