Pytorch GPU顯存充足卻顯示out of memory的解決方式
更新時(shí)間：2020年01月13日 10:14:08 作者：imaginist233
今天小編就為大家分享一篇Pytorch GPU顯存充足卻顯示out of memory的解決方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過來看看吧
今天在測試一個(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
大功告成!
以上這篇Pytorch GPU顯存充足卻顯示out of memory的解決方式就是小編分享給大家的全部內(nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持腳本之家。
相關(guān)文章
python paramiko連接ssh實(shí)現(xiàn)命令
這篇文章主要為大家介紹了python paramiko連接ssh實(shí)現(xiàn)的命令詳解，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2022-07-07
這篇文章主要介紹了Python讀寫unicode文件的方法,涉及Python針對(duì)文件的讀取及編碼操作的相關(guān)技巧,具有一定參考借鑒價(jià)值,需要的朋友可以參考下2015-07-07
Python實(shí)現(xiàn)不一樣的猜數(shù)字游戲的示例代碼
大家知道“猜數(shù)字”這個(gè)游戲嗎？顧名思義就是一個(gè)人想一個(gè)數(shù)字，另一個(gè)人猜。本文就來用Python實(shí)現(xiàn)一款不一樣的猜數(shù)字游戲，感興趣的可以了解一下2023-02-02
Python實(shí)現(xiàn)的遠(yuǎn)程文件自動(dòng)打包并下載功能示例
這篇文章主要介紹了Python實(shí)現(xiàn)的遠(yuǎn)程文件自動(dòng)打包并下載功能,結(jié)合實(shí)例形式分析了Python使用spawn()方法執(zhí)行ssh、scp 命令實(shí)現(xiàn)遠(yuǎn)程文件的相關(guān)操作技巧,需要的朋友可以參考下2019-07-07
Python中可復(fù)用函數(shù)的6種實(shí)踐
為了實(shí)現(xiàn)可維護(hù)性,我們的Python函數(shù)應(yīng)該：小型、只做一項(xiàng)任務(wù)；沒有重復(fù)；有一個(gè)層次的抽象性；有一個(gè)描述性的名字和有少于四個(gè)參數(shù),下面我們就來看看這6個(gè)特性的實(shí)踐吧2023-08-08
這篇文章主要介紹了Python中正則表達(dá)式的詳細(xì)教程,正則表達(dá)式是Python學(xué)習(xí)進(jìn)階當(dāng)中的重要內(nèi)容,需要的朋友可以參考下2021-08-08
python中pyc、?pyd文件及生成使用完整實(shí)例代碼
這篇文章主要介紹了python中pyc、?pyd文件及生成使用的相關(guān)資料,重點(diǎn)講解了如何使用Python解釋器編譯.py文件為.pyc文件,以及如何使用Cython和distutils工具將C/C++代碼編譯為.pyd文件,需要的朋友可以參考下2025-04-04
Python面向?qū)ο缶幊剃P(guān)鍵深度探索類與對(duì)象
這篇文章主要為大家介紹了Python面向?qū)ο缶幊剃P(guān)鍵深度探索類與對(duì)象示例解析，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步，早日升職加薪2023-05-05
Python編程根據(jù)字典列表相同鍵的值進(jìn)行合并
這篇文章主要介紹了來學(xué)習(xí)Python字典列表根據(jù)相同鍵的值進(jìn)行合并的操作方法，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進(jìn)步2021-10-10
Python使用sklearn庫實(shí)現(xiàn)的各種分類算法簡單應(yīng)用小結(jié)
這篇文章主要介紹了Python使用sklearn庫實(shí)現(xiàn)的各種分類算法,結(jié)合實(shí)例形式分析了Python使用sklearn庫實(shí)現(xiàn)的KNN、SVM、LR、決策樹、隨機(jī)森林等算法實(shí)現(xiàn)技巧,需要的朋友可以參考下2019-07-07