TensorFlow下载文件到当前目录

简述

当使用TensorFlow的时候，默认下载文件，如果是windows会进入到C盘的用户界面内。
如果是linux用户，会进入到root目录下。

解决办法

import os import tensorflow as tf tf.enable_eager_execution()train_dataset_url = "http://download.tensorflow.org/data/iris_training.csv"train_dataset_fp = tf.keras.utils.get_file(fname='./'+os.path.basename(train_dataset_url),origin=train_dataset_url, cache_dir='.')print("Local copy of the dataset file: {}".format(train_dataset_fp))

这样就会代码运行的目录下创建一个datasets的目录，下载下来的代码就会放到这里面。

关键点：

• cache_dir='.' 指示目标文件路径。（.表示当前路径）

总结

以上是生活随笔为你收集整理的TensorFlow下载文件到当前目录的全部内容，希望文章能够帮你解决所遇到的问题。

如果觉得生活随笔网站内容还不错，欢迎将生活随笔推荐给好友。