I am trying to use pytorch in AWS lambda but I keep getting this error. I am using
arn:aws:lambda:us-east-1:934676248949:layer:pytorchv1-py36:2
as the layer for pytorch and this is my handler function
import unzip_requirements
import torch
def lambda_handler(event, context):
return "Hello"
As in the function I am just only importing torch and it is giving me this error.
{
"errorMessage": "Unable to import module 'lambda_function': No module named 'torch._C'",
"errorType": "Runtime.ImportModuleError"
}