推 TuCH: 可以先檢查有沒有gpu 有就執行某些code 大部分都一樣 05/07 11:21
→ TuCH: tensorflow 跟 tensorflow-gpu 的code 應該是一樣的 05/07 11:23
推 lemon651: tensorflow 都是一樣的code 05/07 11:43
推 sma1033: model運算那邊可能不一樣,intel CPU是NCHW,Nvidia GPU 05/07 12:26
→ sma1033: 是NHWC,如果model在GPU上train好丟到CPU上跑好像會報錯 05/07 12:27
推 sma1033: code那邊應該要做一些手腳,可能要先判斷硬體種類之類的 05/07 12:31
→ aa12twtwaa: 用gpu train出來的模型可以直接給一般tensorflow用, 05/07 12:31
→ aa12twtwaa: 因為我都是用gpu train完之後丟aws跑 05/07 12:31
推 sma1033: 新版有可能API會自動判斷,我之前用舊的要手動設定NCHW 05/07 12:35
→ sma1033: 設錯了會報Error不給跑 05/07 12:35
→ aa12twtwaa: 這我倒是不清楚了 05/07 12:53
推 sma1033: 像是這個就要設NCHW (tf.nn.conv2d) 05/07 14:43