作者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/cn.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