TensorFlow Model Deployment
Pre-requisites:
Save TF Model
tf_model.save('saved_model/my-model')Create Dockerfile
By default TensorFlow serving looks at /models/model/<integer> directory for saved models. Lets copy our saved model:
RUN mkdir -p /models/model/1
COPY ./saved_model/my-model/ /models/model/1/
EXPOSE 8501
Build & Run Docker Container
docker build -t my-model .docker container run --name my-model -p 8080:8501 -it my-model
Comments
Post a Comment