Docker를 통한 TensorFlow Serving 구축

2384 단어
최근 Docker로 Tensor Flow Serving을 구축했는데 공식 자료를 찾아본 후에 그 문서에 불필요한 절차가 적지 않은 것을 발견하고 한 걸음 한 걸음 배열하여 마침내 더욱 간단한 Docker 이미지 구축 방법을 찾았다.여기에는 두 가지 방법이 있다.
버전 1:
FROM ubuntu:18.04

# Install general packages
RUN apt-get update && apt-get install -y wget && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*
    
# New installation of tensorflow-model-server    
RUN TEMP_DEB="$(mktemp)" \
    && wget -O "$TEMP_DEB" 'http://storage.googleapis.com/tensorflow-serving-apt/pool/tensorflow-model-server-1.8.0/t/tensorflow-model-server/tensorflow-model-server_1.8.0_all.deb' \ 
    && dpkg -i "$TEMP_DEB" \ 
    && rm -f "$TEMP_DEB" \ 
    && mkdir /tmp/model-export    
    
EXPOSE 9000

# Serve the model when the container starts
ENTRYPOINT ["tensorflow_model_server"]
CMD ["--port=9000", "--model_name=model", "--model_base_path=/tmp/model-export"]

버전 2
FROM ubuntu:18.04

# Install general packages
RUN apt-get update && apt-get install -y curl gnupg

# New installation of tensorflow-model-server    
RUN echo "deb [arch=amd64] http://storage.googleapis.com/tensorflow-serving-apt stable tensorflow-model-server tensorflow-model-server-universal" | tee /etc/apt/sources.list.d/tensorflow-serving.list \ 
    && curl https://storage.googleapis.com/tensorflow-serving-apt/tensorflow-serving.release.pub.gpg | apt-key add - \ 
    && apt-get update && apt-get install tensorflow-model-server \ 
    && apt-get clean \ 
    && rm -rf /var/lib/apt/lists/*  \
    && mkdir /tmp/model-export 

EXPOSE 9000

# Serve the model when the container starts
ENTRYPOINT ["tensorflow_model_server"]
CMD ["--port=9000", "--model_name=model", "--model_base_path=/tmp/model-export"]

버전이 만들어지면 Docker 거울이 더 작아지기 때문에 첫 번째 방법을 추천합니다.왜 두 번째 버전이 생겼는지는 공식 문서에서 찾았고 첫 번째 버전은 다른 사람이 제기한 문제에 대한 해답에서 비롯되었다.
위 코드를 dockerfile 파일로 저장하고 docker build 명령을 실행합니다.
docker build -t tensorflow-serving -f dockerfile .

그런 다음 docker run를 통해 컨테이너를 부팅하면 됩니다.
docker run -p 9000:9000 tensorflow-serving

전재 대상:https://www.cnblogs.com/kenwoo/p/9157704.html

좋은 웹페이지 즐겨찾기