0

我正在尝试使用 AI 平台来部署 scikit-learn 管道。我尝试部署的 model.joblib 文件大小为 45 兆字节。

  • 蟒蛇版本:3.7
  • 框架:scikit-learn(==0.20.4)
  • 单核 CPU、四核 CPU(测试版)

我使用以下命令进行部署以及 GUI

 gcloud beta ai-platform versions create v0 \                                                                              
  --model test_watch_model \
  --origin gs://rohan_test_watch_model \
  --runtime-version=1.15 \
  --python-version=3.7 \
  --package-uris=gs://rohan_test_watch_model/train_custom-0.1.tar.gz \
  --framework=scikit-learn \
  --project=xxxx

这是我正在使用的 setup.py 文件,以防问题可能出在库上。

from setuptools import setup

setup(
    name='train_custom',
    version='0.1',
    scripts=[
        # 'train_custom.py',
        # 'data_silo_custom.py',
        # 'dataset_custom.py',
        # 'preprocessor_custom.py'
        'all.py'
    ],
    install_requires=[
        "torch==1.5.1",
        "transformers==3.0.2",
        "farm==0.4.6"
    ]
)

我还尝试从 setup.py 中删除 pytorch 并使用http://storage.googleapis.com/cloud-ai-pytorch/readme.txt中的 torch 1.3,但这给我留下了同样的错误消息。

ERROR: (gcloud.beta.ai-platform.versions.create) Create Version failed. Bad model detected with error: Model requires more memory than allowed. Please try to decrease the model size and re-deploy. If you continue to experience errors, please contact support.
4

0 回答 0