在 AWS lambda 中使用 Pytorch 错误 - “无法导入模块 'lambda_function'：没有名为 'torch._C' 的模块”，

Question

我尝试在 AWS lambda 中使用 pytorch，但不断收到此错误。我用作arn:aws:lambda:us-east-1:934676248949:layer:pytorchv1-py36:2pytorch 的层，这是我的处理函数

import unzip_requirements
import torch



def lambda_handler(event, context):
    
    return "Hello"

正如在函数中一样，我只是导入火炬，它给了我这个错误。

{
  "errorMessage": "Unable to import module 'lambda_function': No module named 'torch._C'",
  "errorType": "Runtime.ImportModuleError"
}

Answer 1

确保您的 AWS Lambda 运行时和容器（使用层时）或 EC2 实例（从 EC2 安装在 EFS 上时）使用相同的 Python 版本。
PyTorch C 扩展是针对特定 Python 版本编译的，差异将导致出现错误。