Amazon web services AWS lambda-sam构建为torch库构建依赖项时返回错误

在本教程之后，我将尝试使用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

---

不幸的是，没有关于错误原因的信息。这与火炬包的尺寸有关吗？如果是这样的话，有什么解决办法吗？

我查看了AWS提供的一个模板，其中他们使用了PyTorch模型。在那里，他们通过

requirements.txt

文件中的以下方法解决问题：

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

---

我不确定，这个版本是否更轻量级，但它可以工作。

你的Lambda函数的python版本是什么？它是python 3.8，我相信你需要先将torch库上传到S3 bucket。然后，您需要在Lambda函数中引用它。对Lambda的依赖关系有一个限制（50MB用于压缩文件，250MB用于解压缩文件）。看见