iT邦幫忙

2021 iThome 鐵人賽

DAY 3
1
AI & Data

機器學習與前端網頁系列 第 3

Day 3 跑一下 Tensorflow 範例

安裝 tensorflow-gpu (如果自己電腦沒有 GPU 就裝 CPU 版)

pip install tensorflow-gpu

跑 tensorflow 的初學者範例。
https://www.tensorflow.org/tutorials/quickstart/beginner

# a02_tf_mnist.py
import tensorflow as tf

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)

參考資料
https://www.tensorflow.org/tutorials/quickstart/beginner


上一篇
Day 2 弄好環境,跑一個範例伺服器
下一篇
Day 4 tensorflow 模型的存取與 mnist 分類
系列文
機器學習與前端網頁30

1 則留言

0
阿瑜
iT邦新手 2 級 ‧ 2021-09-19 22:20:59

友廷老師,
如果有GPU怎麼在Run Code時知道用的是GPU還CPU?

johnting iT邦新手 5 級 ‧ 2021-09-20 01:10:35 檢舉

可以用 以下指令來確定是否有可用的 GPU,有的話它會自動使用。

tf.config.list_physical_devices('GPU')

如果想在 run 期間看硬體使用資訊,可以在使用以下指令開啟log。

tf.debugging.set_log_device_placement(True)

另外,雖然我安裝了 gpu 版本的 tensorflow,
但實際上我電腦 gpu 並沒有支援 cuda...

我要留言

立即登入留言