tensorflow強制使用CPU

Keras若是是使用Theano後端的話,應該是自動不使用GPU只是用CPU的,啓動GPU使用Theano內部命令便可。  對於Tensorflow後端的Keras以及Tensorflow會自動使用可見的GPU,而我須要其必須只運行在CPU上。網上查到三種方法,最後一種比較方便:python 使用tensorflow的 with tf.device('/cpu:0'):函數。簡單操做就是把全部命令
相關文章
相關標籤/搜索