方法1
在users文件夾下新建.keras文件夾,在文件夾下新建keras.json文件,json內(nèi)容為:
{
"image_dim_ordering": "tf",
"epsilon": 1e-07,
"floatx": "float32",
"backend": "tensorflow"
}
這其中的backend可以是tensorflow或者theano,當(dāng)然前提是你已經(jīng)安裝好了。
但是上面這種方式有一個(gè)弊端就是每次當(dāng)你想切換backend的時(shí)候都需要手動(dòng)修改這個(gè)文件,所以有一種能夠動(dòng)態(tài)修改的方式不是更好嗎?請(qǐng)看下面的方法:
方法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加載,設(shè)置order為tensorflow格式。
補(bǔ)充:解決引入keras后出現(xiàn)的Using TensorFlow backend的錯(cuò)誤
問題:
用Anaconda檢索keras-gpu,apply后,理論上應(yīng)該自動(dòng)配好相關(guān)的cuddn和tensorflow-gpu及其相應(yīng)版本,但運(yùn)行程序時(shí)出現(xiàn)如標(biāo)題錯(cuò)誤和一堆報(bào)錯(cuò)。
按網(wǎng)上一堆方法改,都不好用。比如,加os.environ['KERAS_BACKEND']='tensorflow-gpu',更新setup tool等。
解決:
最后誤打誤撞,在Anaconda上修改tensorflow-gpu版本由之前默認(rèn)的1.10改成1.80
![](/d/20211017/d49dcf2ea1ca3d8fe46ee9ea857fc724.gif)
![](/d/20211017/2282296b926aa76ecaeb3b4a18971f61.gif)
python3.6
以上為個(gè)人經(jīng)驗(yàn),希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
您可能感興趣的文章:- keras的get_value運(yùn)行越來越慢的解決方案
- 基于keras中訓(xùn)練數(shù)據(jù)的幾種方式對(duì)比(fit和fit_generator)
- 淺談Keras中fit()和fit_generator()的區(qū)別及其參數(shù)的坑
- Keras保存模型并載入模型繼續(xù)訓(xùn)練的實(shí)現(xiàn)
- TensorFlow2.0使用keras訓(xùn)練模型的實(shí)現(xiàn)
- tensorflow2.0教程之Keras快速入門
- 淺析關(guān)于Keras的安裝(pycharm)和初步理解
- 基于Keras的擴(kuò)展性使用