解決Keras 與 Tensorflow 版本之間的相容性問題
阿新 • • 發佈:2020-02-10
在利用Keras進行實驗的時候,後端為Tensorflow,出現了以下問題:
1. 伺服器端啟用Anaconda環境跑程式時,實驗結果很差。
環境:tensorflow 1.4.0,keras 2.1.5
2. 伺服器端未啟用Anaconda環境跑程式時,實驗結果回到正常值。
環境:tensorflow 1.7.0,keras 2.0.8
3. 自己PC端跑相同程式時,實驗結果回到正常值。
環境:tensorflow 1.6.0,keras 2.1.5
懷疑實驗結果的異常性是由於Keras和Tensorflow版本之間的相容性導致的。查閱網上資料,借鑑他人的經驗,將伺服器端的Anaconda環境配置為:tensorflow 1.4.0,keras 2.0.8,實驗結果恢復了正常。
這裡貼出參考的Keras和Tensorflow版本之間相容性的連結,供參考:https://docs.floydhub.com/guides/environments/。
在配置環境和呼叫庫的時候,需要注意庫之間、與環境之間版本與API的相容性。特別地,cuda版本、cudnn版本、tensorflow版本等。
以上這篇解決Keras 與 Tensorflow 版本之間的相容性問題就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。