如何设置tf.keras.datasets下载CIFAR10数据集的存储目录？

其实你不用额外找其他函数，datasets.cifar10.load_data()本身就支持指定下载路径——它有个容易被忽略的可选参数path，直接传入你想要的目录就行。

给你调整后的代码示例：

from tensorflow.keras import datasets

# 替换成你自己想存储数据集的路径即可
custom_download_path = "./my_cifar10_dataset/"
(train_images, train_labels), (test_images, test_labels) = datasets.cifar10.load_data(path=custom_download_path)

# 归一化操作和之前保持一致
train_images, test_images = train_images / 255.0, test_images / 255.0

顺便补充几个实用细节：

• 要是不指定path参数，TensorFlow默认会把数据集下载到用户目录下的.keras/datasets/文件夹：Windows系统一般是C:\Users\<你的用户名>\.keras\datasets，Linux和macOS则是~/.keras/datasets。
• 如果你指定的路径不存在，TensorFlow会自动帮你创建这个目录，不用手动提前新建。
• 如果你之前已经在默认路径下载过CIFAR10，指定新路径后会重新下载一份到新目录，不会影响或覆盖原来的文件。

内容的提问来源于stack exchange，提问作者VansFannel