Using Pytorch in AWS lambda Error- "Unable to import module 'lambda_function': No module named 'torch._C'",

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