作者ldeathkiller (之之)
看板Python
标题[问题] 在无GPU电脑使用tensorflow-gpu
时间Tue May 7 11:01:31 2019
我知道这问题真的很怪
但有这需求我也很困扰...
想请问各位大大
因为我必须开发同时support GPU以及non-GPU的软体
与此同时我的开发环境没有GPU...
也就是遇到了标题的问题
请问有可能在没有GPU的电脑使用tensorflow-gpu吗?
谢谢各位 QQ
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 61.216.138.92
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/Python/M.1557198093.A.901.html
1F:推 TuCH: 可以先检查有没有gpu 有就执行某些code 大部分都一样 05/07 11:21
2F:→ TuCH: tensorflow 跟 tensorflow-gpu 的code 应该是一样的 05/07 11:23
3F:推 lemon651: tensorflow 都是一样的code 05/07 11:43
4F:推 sma1033: model运算那边可能不一样,intel CPU是NCHW,Nvidia GPU 05/07 12:26
5F:→ sma1033: 是NHWC,如果model在GPU上train好丢到CPU上跑好像会报错 05/07 12:27
6F:推 sma1033: code那边应该要做一些手脚,可能要先判断硬体种类之类的 05/07 12:31
7F:→ aa12twtwaa: 用gpu train出来的模型可以直接给一般tensorflow用, 05/07 12:31
8F:→ aa12twtwaa: 因为我都是用gpu train完之後丢aws跑 05/07 12:31
9F:推 sma1033: 新版有可能API会自动判断,我之前用旧的要手动设定NCHW 05/07 12:35
10F:→ sma1033: 设错了会报Error不给跑 05/07 12:35
11F:→ aa12twtwaa: 这我倒是不清楚了 05/07 12:53
12F:推 sma1033: 像是这个就要设NCHW (tf.nn.conv2d) 05/07 14:43