template.yaml

AWSTemplateFormatVersion: '2010-09-09'
Transform: AWS::Serverless-2016-10-31
Description: >
  pytorch-inference-docker-lambda

  SAM Template for pytorch-inference-docker-lambda

Resources:
  PyTorchInferenceFunction:
    Type: AWS::Serverless::Function # More info about Function Resource: https://github.com/awslabs/serverless-application-model/blob/master/versions/2016-10-31.md#awsserverlessfunction
    Properties:
      PackageType: Image
      MemorySize: 768
      Timeout: 120
    Metadata:
      DockerTag: python3.7-v1
      DockerContext: .
      Dockerfile: Dockerfile

Outputs:
  PyTorchInferenceFunction:
    Description: "PyTorchInference Lambda Function ARN"
    Value: !GetAtt PyTorchInferenceFunction.Arn
  PyTorchInferenceFunctionIamRole:
    Description: "Implicit IAM Role created for PyTorchInference function"
    Value: !GetAtt PyTorchInferenceFunction.Arn