Keras如果是使用Theano后端的话,应该是自动不使用GPU只是用CPU的,启动GPU使用Theano内部命令即可。 对于Tensorflow后端的Keras以及Tensorflow会自动使用可见的GPU,而我需要其必须只运行在CPU上。网上查到三种方法,最后一种方法对我有用,但也对三种都做如下记录: 使用tensorflow的 with tf.device('/cpu:0'):函数。简单
目录 Model 类(函数式 API) Model 类模型方法 compile fit evaluate predict train_on_batch test_on_batch predict_on_batch fit_generator evaluate_generator predict_generator get_layer Model 类(函数式 API)