Skip to content

Repository files navigation

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

Usage

'use strict';//npm install --save gulp gulp-zip gulp-awslambdaconstgulp=require('gulp');constzip=require('gulp-zip');constpath=require('path');constlambda=require('gulp-awslambda');constaws_lamda_tensorflow=require('aws-lambda-tensorflow');constlambda_params={FunctionName: 'addtensorflow',/* Lambda function name */Description: 'My tensorflow lambda function that adds two numbers',//Description for your lambda functionHandler: 'simple_add.lambda_handler',//Assuming you will provide main.py file with a function called handler.MemorySize: 512,Runtime: 'python2.7',Role : 'ROLE_STRING',//eg:'arn:aws:iam::[Account]:role/lambda_basic_execution'Timeout: 50};varopts={region : 'ap-southeast-2'}gulp.task('default',()=>{returngulp.src(['simple_add.py']).pipe(aws_lamda_tensorflow({region : 'ap-southeast-2'}))//Adds all the required files needed to run tensor flow in aws lambda.pipe(zip('archive.zip')).pipe(lambda(lambda_params,opts)).pipe(gulp.dest('dist'));});

API

aws_lamda_tensorflow(options)

options

region(mandatory)

Six lambda available regions.

Type: string
values: ap-northeast-1, ap-southeast-2, eu-central-1, eu-west-1, us-east-1, us-west-2

version (optional)

Use the specified version of tensorflow.

Check this repository for the available versions

Type: string
Default: LATEST

Sample Project

sample-tensorflow-aws-lambda

Contributing

Just give a pull request.

See Creating-Dependency.md.

About

Gulp plugin to deploy tensorflow in aws lambda

Resources

Stars

17 stars

Watchers

1 watching

Forks

Releases

Packages

Used by

Contributors

Languages