剛開始學卷積神經網路 用tensorflow 寫的四層 兩層卷積加池化 兩層Fc 用Spyder CPU內存12G i5 7代 用的mnist數據 電腦總是跑卡 內存佔用100% 有沒有比較好的解決辦法?


  1. 內存佔滿換顯卡也沒用,顯存同一級別基本很少比內存更大的。 一個可能是每次所有樣本一起訓練,建議用分batch訓練。
  2. 如果分batch訓練的話,調整batchsize大小,比如從128調為10,極限為1
  3. 如果還卡,估計就是參數太多,比如一個圖片就50M,那就重新規劃訓練樣本。


換電腦吧


如果能正常運行,那代碼應該沒問題,卡可能是電腦配置的問題。而且對於cnn一般用GPU運行會合理一些。可以試試在智星雲分時租用一下GPU,看看跑的效果。


補充,知乎體,得謝邀,不然顯得沒禮貌

1.有錢方案

換電腦啊,遊戲gpu也行啊,gtx1080ti不算貴,攢一下3k買個二手可以的

2. 沒錢方案

2.1 問室友借一下電腦,訓練的時候搞個分散式cpu小集群

2.2 薅羊毛,使用谷歌的深度學習免費平台

2.3 大家都有提到,batch再減小一點,不要一下子塞那麼多數據進內存

2.4 換mxnet試試,這個框架內存使用大家都贊


使用谷歌的免費GPU是一個不錯的選項


推薦閱讀:
相关文章