keras修改backend的簡單方法

Posted on by

方法1

在users文件夾下新建.keras文件夾，在文件夾下新建keras.json文件，json內容為：

{
    "image_dim_ordering": "tf", 
    "epsilon": 1e-07, 
    "floatx": "float32", 
    "backend": "tensorflow"
}

這其中的backend可以是tensorflow或者theano,當然前提是你已經安裝好瞭。

但是上面這種方式有一個弊端就是每次當你想切換backend的時候都需要手動修改這個文件，所以有一種能夠動態修改的方式不是更好嗎？請看下面的方法：

方法2

import os
os.environ['KERAS_BACKEND'] = 'tensorflow'
import keras.backend as K
K.set_image_dim_ordering('tf')

首先載入os庫，將keras的backend修改為tensorflow，再將keras的backend加載，設置order為tensorflow格式。

補充：解決引入keras後出現的Using TensorFlow backend的錯誤

問題：

用Anaconda檢索keras-gpu，apply後，理論上應該自動配好相關的cuddn和tensorflow-gpu及其相應版本，但運行程序時出現如標題錯誤和一堆報錯。

按網上一堆方法改，都不好用。比如，加os.environ[‘KERAS_BACKEND’]=’tensorflow-gpu’，更新setup tool等。

解決：

最後誤打誤撞，在Anaconda上修改tensorflow-gpu版本由之前默認的1.10改成1.80

python3.6

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。

推薦閱讀：