關(guān)于TensorFlow、Keras、Python版本匹配一覽表

更新時(shí)間：2024年03月28日 09:32:33 作者：許野平

這篇文章主要介紹了關(guān)于TensorFlow、Keras、Python版本匹配一覽表,具有很好的參考價(jià)值,希望對(duì)大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教

TensorFlow、Keras、Python 版本匹配一覽表

興沖沖裝完軟件，發(fā)現(xiàn)運(yùn)行不了，查了下資料，發(fā)現(xiàn)是TensorFlow、Keras、Python 版本匹配問(wèn)題。

這里提供一個(gè)版本匹配清單，需要嚴(yán)格按此標(biāo)準(zhǔn)安裝。

版本匹配清單

Framework	Env name	Description
TensorFlow 2.2	tensorflow-2.2	TensorFlow 2.2.0 + Keras 2.3.1 on Python 3.7.
TensorFlow 2.1	tensorflow-2.1	TensorFlow 2.1.0 + Keras 2.3.1 on Python 3.6.
TensorFlow 2.0	tensorflow-2.0	TensorFlow 2.0.0 + Keras 2.3.1 on Python 3.6.
TensorFlow 1.15	tensorflow-1.15	TensorFlow 1.15.0 + Keras 2.3.1 on Python 3.6.
TensorFlow 1.14	tensorflow-1.14	TensorFlow 1.14.0 + Keras 2.2.5 on Python 3.6.
TensorFlow 1.13	tensorflow-1.13	TensorFlow 1.13.0 + Keras 2.2.4 on Python 3.6.
TensorFlow 1.12	tensorflow-1.12	TensorFlow 1.12.0 + Keras 2.2.4 on Python 3.6.
tensorflow-1.12:py2	TensorFlow 1.12.0 + Keras 2.2.4 on Python 2.
TensorFlow 1.11	tensorflow-1.11	TensorFlow 1.11.0 + Keras 2.2.4 on Python 3.6.
tensorflow-1.11:py2	TensorFlow 1.11.0 + Keras 2.2.4 on Python 2.
TensorFlow 1.10	tensorflow-1.10	TensorFlow 1.10.0 + Keras 2.2.0 on Python 3.6.
tensorflow-1.10:py2	TensorFlow 1.10.0 + Keras 2.2.0 on Python 2.
TensorFlow 1.9	tensorflow-1.9	TensorFlow 1.9.0 + Keras 2.2.0 on Python 3.6.
tensorflow-1.9:py2	TensorFlow 1.9.0 + Keras 2.2.0 on Python 2.
TensorFlow 1.8	tensorflow-1.8	TensorFlow 1.8.0 + Keras 2.1.6 on Python 3.6.
tensorflow-1.8:py2	TensorFlow 1.8.0 + Keras 2.1.6 on Python 2.
TensorFlow 1.7	tensorflow-1.7	TensorFlow 1.7.0 + Keras 2.1.6 on Python 3.6.
tensorflow-1.7:py2	TensorFlow 1.7.0 + Keras 2.1.6 on Python 2.
TensorFlow 1.5	tensorflow-1.5	TensorFlow 1.5.0 + Keras 2.1.6 on Python 3.6.
tensorflow-1.5:py2	TensorFlow 1.5.0 + Keras 2.0.8 on Python 2.
TensorFlow 1.4	tensorflow-1.4	TensorFlow 1.4.0 + Keras 2.0.8 on Python 3.6.
tensorflow-1.4:py2	TensorFlow 1.4.0 + Keras 2.0.8 on Python 2.
TensorFlow 1.3	tensorflow-1.3	TensorFlow 1.3.0 + Keras 2.0.6 on Python 3.6.
tensorflow-1.3:py2	TensorFlow 1.3.0 + Keras 2.0.6 on Python 2.

附上一段測(cè)試程序（鳶尾花分類簡(jiǎn)化版）

這一段代碼不需要準(zhǔn)備數(shù)據(jù)文件，可直接驗(yàn)證是否可以訓(xùn)練模型。

#ex7-2.py
#導(dǎo)入庫(kù)包
import numpy as np
import keras
import matplotlib.pyplot as plt
from keras.models import Sequential
from keras.layers import Dense
#讀入數(shù)據(jù)
train_x = np.array([[1.4, 0.2],
                        [1.7, 0.4],
                        [1.5, 0.4],
                        [2.3, 0.7],
                        [2.7, 1.1],
                        [2.6, 0.9],
                        [4.6, 1.3],
                        [3.5, 1.0],
                        [3.9, 1.2]])
train_y = np.array([[1, 0, 0],
                        [1, 0, 0],
                        [1, 0, 0],
                        [0, 1, 0],
                        [0, 1, 0],
                        [0, 1, 0],
                        [0, 0, 1],
                        [0, 0, 1],
                        [0, 0, 1]])
#搭建模型
model = Sequential()
model.add(Dense(units = 2, input_dim = 2))
#model.add(Dense(units = 2, input_dim = 2, activation = 'sigmoid'))
model.add(Dense(units = 3, activation = 'softmax'))
#編譯模型
model.compile(optimizer = 'adam', loss = 'mse')
#訓(xùn)練模型
model.fit(x = train_x, y = train_y, epochs = 10000)
#保存模型
keras.models.save_model(model, 'iris2.model')