最近在玩強化訓練的時候遇到一點小問題就是
因為模型的深度很深需要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)
這兩個方法都沒效= =
有點難過
不知道版上強者沒有研究過，可以指導一下小弟
非常感謝!!