1. 程式人生 > 程式設計 >解決Keras 與 Tensorflow 版本之間的相容性問題

解決Keras 與 Tensorflow 版本之間的相容性問題

在利用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 版本之間的相容性問題就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。