看板 Python 關於我們 聯絡資訊
我知道這問題真的很怪 但有這需求我也很困擾... 想請問各位大大 因為我必須開發同時support GPU以及non-GPU的軟體 與此同時我的開發環境沒有GPU... 也就是遇到了標題的問題 請問有可能在沒有GPU的電腦使用tensorflow-gpu嗎? 謝謝各位 QQ -- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 61.216.138.92 ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1557198093.A.901.html
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