gRPC 的安装：

$ pip install grpcio

安装 ProtoBuf 相关的 python 依赖库：

$ pip install protobuf

安装 python grpc 的 protobuf 编译工具：

$ pip install grpcio-tools

 # run at root of tensorflow_serving repo

 TARGET_DIR="$1"

 python -m grpc.tools.protoc \

     -I . -I ./tensorflow \

     --python_out "$TARGET_DIR" \

     tensorflow_serving/servables/tensorflow/saved_model_bundle_source_adapter.proto \

     tensorflow_serving/servables/tensorflow/session_bundle_config.proto \

     tensorflow_serving/config/platform_config.proto

 pushd $TARGET_DIR

 touch tensorflow_serving/__init__.py

 touch tensorflow_serving/config/__init__.py

 touch tensorflow_serving/servables/__init__.py

 touch tensorflow_serving/servables/tensorflow/__init__.py

 popd

sh gen-tf-serving-proto-py.sh /tmp

cp -r /tmp/tensorflow_serving .

 # -*- coding: utf-8 -*-

 import tensorflow as tf

 from tensorflow_serving.config import platform_config_pb2

 from tensorflow_serving.servables.tensorflow import session_bundle_config_pb2

 from tensorflow_serving.servables.tensorflow import saved_model_bundle_source_adapter_pb2

 session_config = tf.ConfigProto()

 # config whatever you want

 session_config.gpu_options.allow_growth = True

 session_config.gpu_options.per_process_gpu_memory_fraction = 0.4

 legacy_config=session_bundle_config_pb2.SessionBundleConfig(session_config=session_config)

 adapter = saved_model_bundle_source_adapter_pb2.SavedModelBundleSourceAdapterConfig(legacy_config=legacy_config)

 config_map = platform_config_pb2.PlatformConfigMap()

 config_map.platform_configs['tensorflow'].source_adapter_config.Pack(adapter)

 print(config_map)

 platform_configs {

   key: "tensorflow"

   value {

     source_adapter_config {

       [type.googleapis.com/tensorflow.serving.SavedModelBundleSourceAdapterConfig] {

         legacy_config {

           session_config {

             gpu_options {

               per_process_gpu_memory_fraction: 0.4

               allow_growth: true

             }

           }

         }

       }

     }

   }

 }

 platform_configs {

   key: "tensorflow"

   value {

     source_adapter_config {

       [type.googleapis.com/tensorflow.serving.SavedModelBundleSourceAdapterConfig] {

         legacy_config {

       batching_parameters {

         max_batch_size { value:  }

         batch_timeout_micros { value:  }

         max_enqueued_batches { value:  }

         num_batch_threads { value:  }

           }

           session_config {

             allow_soft_placement: true

             gpu_options {

               per_process_gpu_memory_fraction: 0.4

               allow_growth: true

             }

           }

         }

       }

     }

   }

 }

 platform_configs {

   key: "tensorflow"

   value {

     source_adapter_config {

       type_url: "type.googleapis.com/tensorflow.serving.SavedModelBundleSourceAdapterConfig"

       value: "\302>\017\022\r2\013\t\232\231\231\231\231\231\331? \001"

     }

   }

 }