docker
编写Dockerfile
将Dcokerfile文件置于项目根目录下,文件内容如下
FROM python:3.7
WORKDIR ./docker_demo
ADD . .
RUN pip install -r requirements.txt
CMD ["python", "./src/main.py"]
docker run -t --rm -p 8500:8500 -p 8501:8501 -v "/aa/examples/test:/models" -v /aa/config:/config -e TF_CPP_MIN_VLOG_LEVEL=4 tensorflow/serving:2.11.0-gpu -e MODEL_NAME=test --per_process_gpu_memory_fraction=0.8 --enable_batching=true --model_config_file=/config/model.config > /aa/logs/4.log &
/aa/examples/test:/models 路径的映射
config:{
name:'query_tower'
base_path:'/models/query_tower'
model_platform:'tensorflow'
}