[問題] 在無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)

繼續閱讀