返回
TensorFlow 入门:Ubuntu 16.04 安装和使用指南
见解分享
2024-01-13 21:37:39
TensorFlow,由 Google 打造,是一款开源机器学习软件,专为训练神经网络而生。TensorFlow 的神经网络以有状态数据流图的形式呈现。图中每一个节点代表神经网络在多维数组上执行的操作。通常将这些多维数组称为“张量”,因而得名 TensorFlow。TensorFlow 架构允许…
安装 TensorFlow
在 Ubuntu 16.04 上安装 TensorFlow 非常简单,只需几个简单的步骤:
- 更新软件包列表:
sudo apt-get update
- 安装 TensorFlow:
sudo apt-get install python3-tensorflow
使用 TensorFlow
安装完成后,您可以开始使用 TensorFlow。以下是创建简单神经网络的分步指南:
- 导入 TensorFlow:
import tensorflow as tf
- 定义神经网络:
model = tf.keras.models.Sequential([
tf.keras.layers.Dense(units=10, activation='relu', input_shape=(784,)),
tf.keras.layers.Dense(units=10, activation='relu'),
tf.keras.layers.Dense(units=10, activation='softmax')
])
- 编译神经网络:
model.compile(optimizer='adam', loss='sparse_categorical_crossentropy', metrics=['accuracy'])
- 训练神经网络:
model.fit(x_train, y_train, epochs=10)
- 评估神经网络:
model.evaluate(x_test, y_test)
更多资源
有关 TensorFlow 的更多信息,请访问以下资源: