在 Google AI Platform 上运行tensorflow-cloud作业,作业的入口点如下:
import tensorflow as tf
filename = r'gs://my_bucket_name/hello.txt'
with tf.io.gfile.GFile(filename, mode='w') as w:
w.write("Hello, world!")
with tf.io.gfile.GFile(filename, mode='r') as r:
print(r.read())
作业成功完成,在日志中打印“hello world”。
存储桶和作业都在同一个区域中。
但我在 Cloud Storage 中找不到该文件。它不在那里。我跑了一些其他的测试,tf.io.gfile.listdir
然后我写了一个新文件,然后tf.io.gfile.listdir
我打印了之前和之后,似乎添加了一个文件,但是当我打开云存储时,我在那里找不到它。还能够从存储中读取文件。
我没有收到任何权限错误,正如官方文档所说,AI Platform 已经拥有读取/写入云存储的权限。
这是我的main.py
文件:
import tensorflow_cloud as tfc
tfc.run(
entry_point="run_me.py",
requirements_txt="requirements.txt",
chief_config=tfc.COMMON_MACHINE_CONFIGS['CPU'],
docker_config=tfc.DockerConfig(
image_build_bucket="test_ai_storage"),
)
这是我可以重现问题的最小版本。