方法1
在users文件夾下新建.keras文件夾,在文件夾下新建keras.json文件,json內容為:
1
2
3
4
5
6
|
{ "image_dim_ordering" : "tf" , "epsilon" : 1e - 07 , "floatx" : "float32" , "backend" : "tensorflow" } |
這其中的backend可以是tensorflow或者theano,當然前提是你已經安裝好了。
但是上面這種方式有一個弊端就是每次當你想切換backend的時候都需要手動修改這個文件,所以有一種能夠動態修改的方式不是更好嗎?請看下面的方法:
方法2
1
2
3
4
|
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
以上為個人經驗,希望能給大家一個參考,也希望大家多多支持服務器之家。
原文鏈接:https://blog.csdn.net/linuxwindowsios/article/details/68483618