作者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/m.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