Tensorflow排名库的服务功能

Question

我正在尝试为https://github.com/tensorflow/ranking提供服务功能。我能够对其进行训练以获取数据，现在想使用TensorFlow为其提供服务并创建一个Docker映像以将其托管在Sagemaker上。我在执行服务功能时遇到问题。

我已经设置了用于训练和预测的GPU环境。