首先:(本文僅僅是做筆記,友善之後改環境自己對着)
目前環境
window10 + CUDA10.0 + python3.6 + tensorflow-gpu 2.0.0
總結可以選擇的方法:
1、減低包版本 或者 下載下傳tutorials檔案夾 :從上面的csdn參考部落格擷取安裝url
import tensorflow as tf
(train_images, train_labels), (test_images, test_labels) = tf.keras.datasets.mnist.load_data()
# 轉換成one-hot編碼
y_trainOneHot = tf.keras.utils.to_categorical(train_labels)