[問題] 關於tensorflow object detection

看板DataScience作者 (閒學生)時間6年前 (2018/09/17 00:04), 6年前編輯推噓1(102)
留言3則, 1人參與, 6年前最新討論串1/1
(如果用不到以下框架,如觀念問題請自行視情況刪除作業系統,使用工具等項目) 作業系統:(ex:mac,win10,win8,win7,linux,etc...) windows 10 問題類別:(ex:ML,SVM,RL,DL,RNN,CNN,NLP,BD,Vis,etc...) 環境問題 使用工具:(ex:python,Java,C++,tensorflow,keras,MATLAB,etc...) tensorflow 問題內容: 因為最近有需要,所以想要嘗試在window上 使用tensorflow object detection API下去訓練 只是遇到一個狀況 GPU的記憶體會被占用 但是卻沒有進入運算 然後CPU卻跑到100% 而這是我之前在ubuntu上沒有碰過的 原本以為是cuda問題重新安裝一次後還是一樣 目前是使用 python 3.5.5 tensorflow-gpu 1.9 cuda 9.0 cudnn7.2 nvidia drive 399.24X 不知道有沒有大大可以給點方向 感謝! -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 114.39.117.28 ※ 文章網址: https://www.ptt.cc/bbs/DataScience/M.1537113887.A.9F7.html ※ 編輯: owem0410 (114.39.117.28), 09/17/2018 00:07:54

09/17 00:34, 6年前 , 1F
看程式碼有沒有把device跑到cpu去了
09/17 00:34, 1F

09/17 00:36, 6年前 , 2F
然後先確認cpu 100%是卡在訓練還是前處理
09/17 00:36, 2F

09/17 00:37, 6年前 , 3F
gpu占用是只要你叫seesion出來就會占用 不管有沒有run
09/17 00:37, 3F
結果問題出現在更基本的地方... 我沒轉成功tfrecord裡面是空的 當然都跑不過但是因為環境不熟悉所以都沒注意到 真的耍蠢了 ※ 編輯: owem0410 (114.39.117.162), 09/17/2018 23:13:22
文章代碼(AID): #1RddyVdt (DataScience)
文章代碼(AID): #1RddyVdt (DataScience)