【问题标题】:AWS lambda - sam build returns error when building dependencies for torch libraryAWS lambda - sam build 在为火炬库构建依赖项时返回错误
【发布时间】:2021-05-27 21:20:34
【问题描述】:

按照this 教程,我正在尝试使用 pytorch 模型。我按预期设置了 Dockerfile,当我运行 sam build 时,除了要求之外,一切正常。它返回以下错误消息:

Collecting torch==1.8.1
  Downloading torch-1.8.1-cp38-cp38-manylinux1_x86_64.whl (804.1 MB)

Build Failed
Error: HelloWorldFunction failed to build: The command '/bin/sh -c python3.8 -m pip install -r requirements.txt -t .' returned a non-zero code: 137

很遗憾,没有关于导致错误的原因的信息。这与torch包的大小有关吗?如果是这样,有什么解决方法吗?

【问题讨论】:

  • 您的 Lambda 函数上的 python 版本是什么?
  • 是python 3.8
  • 我相信您需要先将torch lib上传到S3存储桶。然后,您需要在 Lambda 函数中引用它。对 Lambda 的依赖项有限制(压缩文件为 50MB,解压缩文件为 250MB)。见docs.aws.amazon.com/lambda/latest/dg/gettingstarted-limits.html

标签: amazon-web-services deployment aws-lambda aws-sam aws-sam-cli


【解决方案1】:

我查看了 AWS 提供的模板,他们在其中使用了 PyTorch 模型。在那里,他们通过requirements.txt 文件中的以下方法解决了这个问题:

-f https://download.pytorch.org/whl/torch_stable.html
torch==1.8.0+cpu

我不确定,如果这个版本更轻量级,但它可以工作。

【讨论】:

    猜你喜欢
    • 1970-01-01
    • 2017-12-24
    • 2022-01-09
    • 2014-12-25
    • 2017-12-12
    • 1970-01-01
    • 2019-12-31
    • 1970-01-01
    • 2013-04-14
    相关资源
    最近更新 更多