Keras以及Tensorflow強制使用CPU,GPU

Keras若是是使用Theano後端的話,應該是自動不使用GPU只是用CPU的,啓動GPU使用Theano內部命令便可。  對於Tensorflow後端的Keras以及Tensorflow會自動使用可見的GPU,而我須要其必須只運行在CPU上。網上查到三種方法,最後一種方法對我有用,但也對三種都作以下記錄:python 使用tensorflow的 with tf.device('/cpu:0'):
相關文章
相關標籤/搜索