[问题] 在无GPU电脑使用tensorflow-gpu ldeathkiller PTT批踢踢实业坊

楼主: ldeathkiller (之之) 2019-05-07 11:01:31

我知道这问题真的很怪
但有这需求我也很困扰...
想请问各位大大
因为我必须开发同时support GPU以及non-GPU的软件
与此同时我的开发环境没有GPU...
也就是遇到了标题的问题
请问有可能在没有GPU的电脑使用tensorflow-gpu吗?
谢谢各位 QQ

作者: TuCH (谬客) 2019-05-07 11:21:00

可以先检查有没有gpu 有就执行某些code 大部分都一样tensorflow 跟 tensorflow-gpu 的code 应该是一样的

作者: lemon651 (小明) 2019-05-07 11:43:00

tensorflow 都是一样的code

作者: sma1033 (死马) 2019-05-07 12:26:00

model运算那边可能不一样，intel CPU是NCHW，Nvidia GPU是NHWC，如果model在GPU上train好丢到CPU上跑好像会报错code那边应该要做一些手脚，可能要先判断硬件种类之类的

作者: aa12twtwaa (Tohka) 2019-05-07 12:31:00

用gpu train出来的模型可以直接给一般tensorflow用，因为我都是用gpu train完之后丢aws跑

作者: sma1033 (死马) 2019-05-07 12:35:00

新版有可能API会自动判断，我之前用旧的要手动设定NCHW设错了会报Error不给跑

作者: aa12twtwaa (Tohka) 2019-05-07 12:53:00

这我倒是不清楚了

作者: sma1033 (死马) 2019-05-07 14:43:00

像是这个就要设NCHW (tf.nn.conv2d)

继续阅读