Skip to content

Latest commit

 

History

History
56 lines (36 loc) · 1.52 KB

File metadata and controls

56 lines (36 loc) · 1.52 KB

RunPod | Python Library

CI | Code Quality   CI | Unit Tests

Official Python library for RunPod API & SDK.

Table of Contents

Installation

pip install runpod

API Language Library

When interacting with the RunPod API you can use this library to make requests to the API.

import runpod

runpod.api_key = "xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"

SDK - Serverless Worker

This python package can also be used to create a serverless worker that can be deployed to RunPod.

Quick Start

Create an python script in your project that contains your model definition and the RunPod worker start code. Run this python code as your default container start command:

import runpod

MODEL = 'YOUR_MODEL'

def run(job):
    # Your inference code here
    return MODEL.predict(job.input)

runpod.serverless.start({"handler": run})

Add the env variables found in serverless-worker to your project.