首先:(本文仅仅是做笔记,方便之后改环境自己对着)
当前环境
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)