App下載

keras怎么修改backend?

猿友 2021-08-05 13:36:23 瀏覽數(shù) (1808)
反饋

在使用Keras的backend的時候可能會遇到一些報錯。這時候我們就需要想辦法修改backend。那么Keras修改backend怎么實現(xiàn)呢,接下來這篇文章帶你了解。

方法1

在users文件夾下新建.keras文件夾,在文件夾下新建keras.json文件,json內容為:

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

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

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

方法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后出現(xiàn)的Using TensorFlow backend的錯誤

問題:

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

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

解決:

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


以上就是Keras修改backend的全部內容,希望能給大家一個參考,也希望大家多多支持W3Cschool


0 人點贊