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