1. 程式人生 > 程式設計 >Keras 切換後端方式(Theano和TensorFlow)

Keras 切換後端方式(Theano和TensorFlow)

實驗室新裝了keras,發現keras預設後端是tensorflow,想換回theano,看了官方文件也沒搞懂,最終搞定,很簡單。

中文文件的描述:keras中文文件,切換後端

其實就是在C:\Users\75538(75538是我的windos使用者名稱字,找你對應的使用者名稱就行)下有個資料夾.keras,裡面有keras.json檔案,改一下里面的內容就好了,如果沒有資料夾和檔案,手動建立就行。

用theano的話,keras.json寫入

{
 "image_dim_ordering": "th","epsilon": 1e-07,"floatx": "float32","backend": "theano"
}

用thesorflow的話,keras.json寫入

{
 "image_dim_ordering": "tf","backend": "tensorflow"
}

最後儲存就可以了

補充知識:anaconda2-keras安裝;keras後端修改

一、anaconda2-keras安裝

在anaconda2平臺下,安裝keras庫。記錄一下,主要是我一開始覺得很簡單,pip一下就行了,結果我用pip install keras安裝時,就遇到了這樣一個問題:ImportError: cannot import name np_utils。

解決方案:

That tutorial was written on June 9th,2016.Keras 2 was released in March 2017. Try installing the old version,using pip installkeras==1.2.2.

總結一下就是:

1.先裝上anaconda2_python2.7

2.再在anancona prompt中用 pip install keras==1.2.2 裝Keras

3.spyder中輸入 import Keras,檢查是否安裝成功

二、keras後端修改

程式碼用的後端是'theano',故要修改後端

將C:\Anaconda2\Lib\site-packages\keras\backend\__init__.py的line 27修改

# Default backend: TensorFlow.
#_BACKEND = 'tensorflow'
_BACKEND = 'theano'

然後,python-> import keras,檢查是否修改成功。

以上這篇Keras 切換後端方式(Theano和TensorFlow)就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。