AWS Lambda 火炬 (python) 失败

AWS Lambda fails for torch (python)

我想做的就是为 Places365 创建一个 Flask 应用程序,然后使用 API 将其部署为 AWS Lambda。虽然在我的 EC-2 实例上一切正常,但 Lambda 总是失败 "No module named 'torch': ModuleNotFoundError" 错误。

最初,当我尝试将 torch 作为我的虚拟环境的一部分时,Lambda 一直失败并出现 "No space left" 错误。因此,我从我的虚拟环境中卸载了 torch,重新部署了函数并将 PyTorch 层 (arn:aws:lambda:us-east-1:934676248949:layer:pytorchv1-py36:2) 添加到函数中。尽管如此,它仍然失败并显示 "No module named 'torch': ModuleNotFoundError" 错误

此外,我使用 Zappa 进行 Lambda 部署

如果有人可以分享他们将 torch 部署到 Lambda 的经验,那就太好了

我能够修复它。下面是我做的

我使用的pytorch层的ARN: arn:aws:lambda:us-east-1:934676248949:layer:pytorchv1-py36:2

将以下代码添加到我的 python Lambda 函数

  sys.path.insert(1, '/opt')
  import unzip_requirements
  import torch