gpt4 book ai didi

docker - docker如何运行tensorflow-serving及其模型?

转载 作者:行者123 更新时间:2023-12-02 19:49:23 25 4
gpt4 key购买 nike

问题是,我想通过docker运行tensorflow-serving及其模型,但似乎
docker将使tensorflow-serving无法加载该文件夹。
在容器本身中,它可以轻松加载模型:

tensorflow_model_server --port=9000 --model_name=1 --model_base_path=/models/

但是使用docker,它无法加载模型:
docker run --name tfx-test -p 8501:8501 -v "/models:/models" -e MODEL_NAME=5 -t tensorflow/serving

结果:
E tensorflow_serving/sources/storage_path/file_system_storage_path_source.cc:362] FileSystemStoragePathSource encountered a filesystem access error: Could not find base path /models/5 for servable 5

模型文件:
|--models
|-- 1
|-- assets
|-- saved_model.pb
`-- variables
|-- variables.data-00000-of-00002
|-- variables.data-00001-of-00002
`-- variables.index

最佳答案

docker run命令更改为:

docker run -p 8501:8501 --mount type=bind,\
source=/models/1,\
target=/models/my_model/1 \
-e MODEL_NAME=my_model -t tensorflow/serving

并且,您可以在以下位置看到模型元数据: http://localhost:8501/v1/models/my_model/metadata(假设您正在从 localhost运行docker)

关于docker - docker如何运行tensorflow-serving及其模型?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/61850960/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com