keras修改backend的簡單方法
方法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)安裝好了。
但是上面這種方式有一個弊端就是每次當(dāng)你想切換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加載,設(shè)置order為tensorflow格式。
補充:解決引入keras后出現(xiàn)的Using TensorFlow backend的錯誤
問題:
用Anaconda檢索keras-gpu,apply后,理論上應(yīng)該自動配好相關(guān)的cuddn和tensorflow-gpu及其相應(yīng)版本,但運行程序時出現(xiàn)如標(biāo)題錯誤和一堆報錯。
按網(wǎng)上一堆方法改,都不好用。比如,加os.environ['KERAS_BACKEND']='tensorflow-gpu',更新setup tool等。
解決:
最后誤打誤撞,在Anaconda上修改tensorflow-gpu版本由之前默認(rèn)的1.10改成1.80


python3.6
以上為個人經(jīng)驗,希望能給大家一個參考,也希望大家多多支持本站。
版權(quán)聲明:本站文章來源標(biāo)注為YINGSOO的內(nèi)容版權(quán)均為本站所有,歡迎引用、轉(zhuǎn)載,請保持原文完整并注明來源及原文鏈接。禁止復(fù)制或仿造本網(wǎng)站,禁止在非maisonbaluchon.cn所屬的服務(wù)器上建立鏡像,否則將依法追究法律責(zé)任。本站部分內(nèi)容來源于網(wǎng)友推薦、互聯(lián)網(wǎng)收集整理而來,僅供學(xué)習(xí)參考,不代表本站立場,如有內(nèi)容涉嫌侵權(quán),請聯(lián)系alex-e#qq.com處理。
關(guān)注官方微信