[問題] 在無GPU電腦使用tensorflow-gpu

看板Python作者 (之之)時間6年前 (2019/05/07 11:01), 編輯推噓6(606)
留言12則, 4人參與, 6年前最新討論串1/1
我知道這問題真的很怪 但有這需求我也很困擾... 想請問各位大大 因為我必須開發同時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

05/07 11:21, 6年前 , 1F
可以先檢查有沒有gpu 有就執行某些code 大部分都一樣
05/07 11:21, 1F

05/07 11:23, 6年前 , 2F
tensorflow 跟 tensorflow-gpu 的code 應該是一樣的
05/07 11:23, 2F

05/07 11:43, 6年前 , 3F
tensorflow 都是一樣的code
05/07 11:43, 3F

05/07 12:26, 6年前 , 4F
model運算那邊可能不一樣,intel CPU是NCHW,Nvidia GPU
05/07 12:26, 4F

05/07 12:27, 6年前 , 5F
是NHWC,如果model在GPU上train好丟到CPU上跑好像會報錯
05/07 12:27, 5F

05/07 12:31, 6年前 , 6F
code那邊應該要做一些手腳,可能要先判斷硬體種類之類的
05/07 12:31, 6F

05/07 12:31, 6年前 , 7F
用gpu train出來的模型可以直接給一般tensorflow用,
05/07 12:31, 7F

05/07 12:31, 6年前 , 8F
因為我都是用gpu train完之後丟aws跑
05/07 12:31, 8F

05/07 12:35, 6年前 , 9F
新版有可能API會自動判斷,我之前用舊的要手動設定NCHW
05/07 12:35, 9F

05/07 12:35, 6年前 , 10F
設錯了會報Error不給跑
05/07 12:35, 10F

05/07 12:53, 6年前 , 11F
這我倒是不清楚了
05/07 12:53, 11F

05/07 14:43, 6年前 , 12F
像是這個就要設NCHW (tf.nn.conv2d)
05/07 14:43, 12F
文章代碼(AID): #1SqFKDa1 (Python)
文章代碼(AID): #1SqFKDa1 (Python)