search

guoqiangqi / PFLD

Implementation of PFLD A Practical Facial Landmark Detector , reference to https://arxiv.org/pdf/1902.10859.pdf
627 stars 166 forks source link

tensorflow gpu是1.6.0版本的，cuda是9.0版本的，尝试训练GPU，可一直失败，是版本号对不上嘛 #10

Closed github-luffy closed 5 years ago

guoqiangqi commented 5 years ago

一直失败指什么情况？

github-luffy commented 5 years ago

使用不了gpu训练，一直在用cpu跑

guoqiangqi commented 5 years ago

我在训练时根据GT计算euler angles，计算Euler angles主要是cpu完成，所以gpu有用到，但是一直跳出tf的session，所以gpu利用率不高，且在cpu上消耗过多时间，将计算euler angles 放在预处理上就可以。

github-luffy commented 5 years ago

好的，你用gpu训练1000epoch大概用了多久的时间？我算了一下，大概要用5天多的时间。

guoqiangqi commented 5 years ago

data/log0.txt可以看

425418567@qq.com

发件人： 叶修强 发送时间： 2019-06-24 15:52 收件人： guoqiangqi/PFLD 抄送： Guoqiang QI; Comment 主题： Re: [guoqiangqi/PFLD] tensorflow gpu是1.6.0版本的，cuda是9.0版本的，尝试训练GPU，可一直失败，是版本号对不上嘛 (#10) 好的，你用gpu训练1000epoch大概用了多久的时间？我算了一下，大概要用5天多的时间。 — You are receiving this because you commented. Reply to this email directly, view it on GitHub, or mute the thread.

github-luffy commented 5 years ago

好的，谢谢你