TensorFlow Hub缓存模型-加载时拒绝权限

时间:2018-12-12 11:36:47

标签: tensorflow anaconda

我正在尝试将TensorFlow模块保存到磁盘上,以避免每次使用时都下载它。

我在这里了解了有关缓存模块的信息:https://www.tensorflow.org/hub/basics

$ export TFHUB_CACHE_DIR=/tf_models
$ echo $TFHUB_CACHE_DIR
/tf_models

因此设置了环境变量,我也将其添加到.bashrc中,并用source重新加载了.bashrc。

在python中:

import tensorflow_hub as hub
embed = hub.Module("https://tfhub.dev/google/universal-sentence-encoder/2")


~/anaconda3/envs/tf-gpu/lib/python3.6/site-packages/tensorflow/python/framework/errors_impl.py in __exit__(self, type_arg, value_arg, traceback_arg)
    526             None, None,
    527             compat.as_text(c_api.TF_Message(self.status.status)),
--> 528             c_api.TF_GetCode(self.status.status))
    529     # Delete the underlying status object from memory otherwise it stays alive
    530     # as there is a reference to status from this from the traceback due to

PermissionDeniedError: /tf_models; Permission denied

默认为TFHUB_CACHE_DIR时,我可以很好地运行集线器模块。

为什么我的权限被拒绝?

1 个答案:

答案 0 :(得分:0)

已通过删除/符号来解决,例如:

export TFHUB_CACHE_DIR=tf_models