Re: [問題] tensorflow- 訓練在gpu 預測在cpu
※ 引述《snakei14702 (阿中是我們的神)》之銘言：
: 最近在玩強化訓練的時候遇到一點小問題就是
: 因為模型的深度很深需要GPU來跑
: 但是在run env 計算(中間有用到model.predict)一些東西的時候
: 如果用GPU跑就顯得太慢沒效率...
: 所以想要請教一下各位有沒有辦法在fit model時用GPU
: 然後用model.prict的時候用cpu呢?
: 簡化的代碼如下:
: for i in range(1000):
: model.fit(x,y)
: model.predict(x)
: 基本上我有用過
: for i in range(1000):
: os.environ["CUDA_VISIBLE_DEVICES"]="0"
: model.fit(x,y)
: os.environ["CUDA_VISIBLE_DEVICES"]="-1"
: model.predict(x)
: and
: for i in range(1000):
: with tf.device('/GPU:0'):
: model.fit(x,y)
: with tf.device('/CPU:0'):
: model.predict(x)
: 這兩個方法都沒效= =
: 有點難過
: 不知道版上強者沒有研究過，可以指導一下小弟
: 非常感謝!!
謝謝各位強者的幫忙
後來研究一下
用@tf.fuction修飾
然後用model(x)取代model.predict(x)
就可以跑的爆炸快了！！！
以前覺得@tf.fuction好像也沒多厲害
不過在RL玩遊戲的時候整個讓我很驚艷！
再次感謝各位的幫忙！
--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 211.23.79.225 (臺灣)
※ 文章網址: https://www.ptt.cc/bbs/Python/M.1664046699.A.23A.html
推
09/26 09:29,
2年前
, 1F
09/26 09:29, 1F
討論串 (同標題文章)
本文引述了以下文章的的內容：
完整討論串 (本文為第 2 之 2 篇)：
Python 近期熱門文章
PTT數位生活區 即時熱門文章