如何在自定义预测例程上安装 lightgbm？

Question

我已经训练了一个 LightGBM 模型，现在想要创建一个自定义预测例程以部署在 google aiplatform 上。

加载模型文件时，会抛出未安装 lightgbm 的错误。我在 setup.py 中使用了下面的代码，但它不起作用。

from setuptools import setup, find_packages
DEPENDENCIES = ['lightgbm==3.1.1']

setup(
      name='my_custom_code',
      version='0.4',
      install_requires=DEPENDENCIES,
      include_package_date=True,
      packages=find_packages(),
      description='custom code.',
      scripts=['predictor.py', 'preprocess.py'])

Answer 1

为了实现您想要实现的目标，您需要创建一个包含预安装的 lightGBM 的自定义容器，在这里您可以找到有关如何构建和推送自定义容器映像的分步指南。