最近在玩强化训练的时候遇到一点小问题就是
因为模型的深度很深需要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)
这两个方法都没效= =
有点难过
不知道版上强者没有研究过，可以指导一下小弟
非常感谢!!