婷婷综合国产,91蜜桃婷婷狠狠久久综合9色 ,九九九九九精品,国产综合av

主頁 > 知識庫 > 粗暴解決CUDA out of memory的問題

粗暴解決CUDA out of memory的問題

熱門標簽:沈陽人工外呼系統價格 富錦商家地圖標注 江西省地圖標注 如何申請400電話費用 沈陽防封電銷卡品牌 外呼系統哪些好辦 池州外呼調研線路 沈陽外呼系統呼叫系統 武漢外呼系統平臺

小渣渣復現大佬project發現GPU跑不動,出現如下報錯:

RuntimeError: CUDA out of memory.

看下來最簡單粗暴方法就是減少batch_size,慢是慢了不止一點點但至少跑得動了!

補充:Pytorch GPU顯存充足卻顯示out of memory解決辦法

今天在測試一個pytorch代碼的時候顯示顯存不足,但是這個網絡框架明明很簡單,用CPU跑起來都沒有問題,GPU卻一直提示out of memory.

在網上找了很多方法都行不通,最后我想也許是pytorch版本的問題,原來我的pytorch版本是0.4.1,于是我就把這個版本卸載,然后安裝了pytorch1.1.0,程序就可以神奇的運行了,不會再有OOM的提示了。雖然具體原因還不知道為何,這里還是先mark一下,

具體過程如下:

卸載舊版本pytorch:

conda uninstall pytorch

安裝pytorch1.1.0,按照官網上的辦法,我的CUDA版本是9.0:

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

大功告成!

以上為個人經驗,希望能給大家一個參考,也希望大家多多支持腳本之家。

您可能感興趣的文章:
  • PyTorch CUDA環境配置及安裝的步驟(圖文教程)
  • 解決PyTorch與CUDA版本不匹配的問題
  • 完美解決torch.cuda.is_available()一直返回False的玄學方法
  • 詳解win10下pytorch-gpu安裝以及CUDA詳細安裝過程
  • 如何解決.cuda()加載用時很長的問題

標簽:黑龍江 通遼 呂梁 銅川 常德 阿里 潛江 株洲

巨人網絡通訊聲明:本文標題《粗暴解決CUDA out of memory的問題》,本文關鍵詞  粗暴,解決,CUDA,out,memory,;如發現本文內容存在版權問題,煩請提供相關信息告之我們,我們將及時溝通與處理。本站內容系統采集于網絡,涉及言論、版權與本站無關。
  • 相關文章
  • 下面列出與本文章《粗暴解決CUDA out of memory的問題》相關的同類信息!
  • 本頁收集關于粗暴解決CUDA out of memory的問題的相關信息資訊供網民參考!
  • 推薦文章
    主站蜘蛛池模板: 二连浩特市| 壶关县| 元氏县| 措美县| 新乐市| 蒙自县| 卢氏县| 清镇市| 利辛县| 德令哈市| 桂林市| 洛南县| 建始县| 黄平县| 明光市| 礼泉县| 常熟市| 开江县| 宁明县| 仙居县| 南木林县| 名山县| 嘉禾县| 乐至县| 织金县| 邓州市| 武义县| 乌拉特后旗| 宜都市| 霞浦县| 门头沟区| 毕节市| 常熟市| 会同县| 福泉市| 沭阳县| 陕西省| 南充市| 丁青县| 清水县| 青铜峡市|