TensorFlow Serving安装

TensorFlow Serving可以使用apt-get或Docker安装。在生产环境中,推荐 使用Docker部署TensorFlow Serving 。不过此处出于教学目的,介绍依赖环境较少的 apt-get安装

警告

软件的安装方法往往具有时效性,本节的更新日期为2019年8月。若遇到问题,建议参考 TensorFlow网站上的最新安装说明 进行操作。

首先设置安装源:

  1. # 添加Google的TensorFlow Serving源
  2. echo "deb [arch=amd64] http://storage.googleapis.com/tensorflow-serving-apt stable tensorflow-model-server tensorflow-model-server-universal" | sudo tee /etc/apt/sources.list.d/tensorflow-serving.list
  3. # 添加gpg key
  4. curl https://storage.googleapis.com/tensorflow-serving-apt/tensorflow-serving.release.pub.gpg | sudo apt-key add -

更新源后,即可使用apt-get安装TensorFlow Serving

  1. sudo apt-get update
  2. sudo apt-get install tensorflow-model-server

提示

在运行curl和apt-get命令时,可能需要设置代理。

curl设置代理的方式为 -x 选项或设置 http_proxy 环境变量,即

  1. export http_proxy=http://代理服务器IP:端口

  1. curl -x http://代理服务器IP:端口 URL

apt-get设置代理的方式为 -o 选项,即

  1. sudo apt-get -o Acquire::http::proxy="http://代理服务器IP:端口" ...

Windows 10下,可以在 Linux子系统(WSL) 内使用相同的方式安装TensorFlow Serving。