Tensorflow的安装和开始

本博客最新更新日期： 2019-11-22

安装

请到这里查看对应的wheel包的地址。拉到最底下：

选择自己对应系统，对应python，对应CPU/GPU的版本。

我的环境是Windows 10.从上面看到windows的只有python3.6的版本（我试过了，用python3.7会报错）。作为新手，下载CPU only版本会减少一开始犯错误的次数。

然后记下刚才的地址，打开终端

pip install https://storage.googleapis.com/tensorflow/windows/cpu/tensorflow-2.0.0-cp36-cp36m-win_amd64.whl

等待安装完成就好了。

开始

书写如下脚本

from __future__ import absolute_import, division, print_function, unicode_literals
import tensorflow as tf
import os
os.environ["TF_CPP_MIN_LOG_LEVEL"]='2' # 只显示 warning 和 Error

mnist = tf.keras.datasets.mnist

(x_train, y_train), (x_test, y_test) = mnist.load_data()
x_train, x_test = x_train / 255.0, x_test/255.0

model = tf.keras.models.Sequential([
  tf.keras.layers.Flatten(input_shape=(28, 28)),
  tf.keras.layers.Dense(128, activation='relu'),
  tf.keras.layers.Dropout(0.2),
  tf.keras.layers.Dense(10, activation='softmax')
])

model.compile(optimizer='adam',
              loss='sparse_categorical_crossentropy',
              metrics=['accuracy'])

model.fit(x_train, y_train, epochs=5)

model.evaluate(x_test,  y_test, verbose=2)

然后运行即可。这段程序会自动下载mnist数据集到本地，进行最基本的训练。

注意第四行：你可以选择把它注释掉，然后它会打印出如下信息：

这段红字是说你的CPU支持AVX2指令集加速，但是你下载的编译的tensorflow不支持使用这个指令集。

这是一条信息，不影响程序运行。最终部分打印结果如下：