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"
}