Keras以及Tensorflow强制使用CPU,GPU

Keras若是是使用Theano后端的话,应该是自动不使用GPU只是用CPU的,启动GPU使用Theano内部命令便可。  对于Tensorflow后端的Keras以及Tensorflow会自动使用可见的GPU,而我须要其必须只运行在CPU上。网上查到三种方法,最后一种方法对我有用,但也对三种都作以下记录:python 使用tensorflow的 with tf.device('/cpu:0'):
相关文章
相关标签/搜索