aws-lambda-tensorflow

aws-lambda-tensorflow

A gulp plugin to deploy python tensorflow in aws lambda.

Prerequisites

This plugin assumes you have aws-cli installed and configured with proper access rights to use aws lambda.

Installation

npm install --save-dev aws-lambda-tensorflow

'use strict';
//npm install --save gulp gulp-zip gulp-awslambda
const gulp   = require('gulp');
const zip    = require('gulp-zip');
const path   = require('path');
const lambda = require('gulp-awslambda');
const aws_lamda_tensorflow = require('aws-lambda-tensorflow');

const lambda_params  = {
    FunctionName: 'mylambdafunction', /* Lambda function name */
    Description: 'My tensorflow lambda function that adds two numbers', //Description for your lambda function
    Handler: 'main.handler', //Assuming you will provide main.py file with a function called handler.
    MemorySize: 128,
    Runtime: 'python2.7',
    Role : 'ROLE_STRING',//eg:'arn:aws:iam::[Account]:role/lambda_basic_execution'
    Timeout: 10
};

gulp.task('default', () => {
    return gulp.src(['main.py'])
                .pipe(aws_lamda_tensorflow()) //Adds all the required files needed to run tensor flow in aws lambda
                .pipe(zip('archive.zip'))
                .pipe(lambda(lambda_params, {}))
                .pipe(gulp.dest('dist'));
});

Sample Project

sample-tensorflow-aws-lambda