代码之家  ›  专栏  ›  技术社区  ›  Joel Holmes

TensorFlow服务S3和Docker

  •  -1
  • Joel Holmes  · 技术社区  · 5 年前

    我试图找到一种使用TensorFlow服务的方法,能够添加新模型和新版本的模型。我可以把TensorFlow指向S3桶吗?

    我还需要它作为一个容器运行?这是可能的,还是需要实现另一个程序来下拉模型并将其添加到共享卷,并要求TensorFlow更新文件系统中的模型?

    还是我需要建立自己的Docker形象,才能从S3中提取内容?

    1 回复  |  直到 5 年前
        1
  •  1
  •   Joel Holmes    5 年前

    我发现我可以使用 TF S3 连接信息(即使在TF服务Docker容器中没有概述)。Docker运行命令示例:

    docker run -p 8501:8501 -e AWS_ACCESS_KEY_ID=$AWS_ACCESS_KEY_ID -e AWS_SECRET_ACCESS_KEY=$AWS_SECRET_ACCESS_KEY -e MODEL_BASE_PATH=s3://path/bucket/models -e MODEL_NAME=model_name -e S3_ENDPOINT=s3.us-west-1.amazonaws.com -e AWS_REGION=us-west-1 -e TF_CPP_MIN_LOG_LEVEL=3 -t tensorflow/serving
    

    注释 因此设置了日志级别 bug