作者owem0410 (閒學生)
看板DataScience
標題[問題] 關於tensorflow object ddetection
時間Mon Sep 17 00:04:44 2018
(如果用不到以下框架，如觀念問題請自行視情況刪除作業系統,使用工具等項目)
作業系統:(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://webptt.com/m.aspx?n=bbs/DataScience/M.1537113887.A.9F7.html
※ 編輯: owem0410 (114.39.117.28), 09/17/2018 00:07:54
1F：推 germun: 看程式碼有沒有把device跑到cpu去了 09/17 00:34
2F：→ germun: 然後先確認cpu 100%是卡在訓練還是前處理 09/17 00:36
3F：→ germun: gpu占用是只要你叫seesion出來就會占用 不管有沒有run 09/17 00:37
結果問題出現在更基本的地方...
我沒轉成功tfrecord裡面是空的
當然都跑不過但是因為環境不熟悉所以都沒注意到
真的耍蠢了
※ 編輯: owem0410 (114.39.117.162), 09/17/2018 23:13:22