title: “Modal Serverless Gpu — Serverless GPU cloud platform for running ML workloads” sidebar_label: “Modal Serverless Gpu” description: “Serverless GPU cloud platform for running ML workloads”

{/* 本页面由 website/scripts/generate-skill-docs.py 根据技能对应的 SKILL.md 文件自动生成。请直接编辑源文件 SKILL.md，而非此页面。 */}

Modal Serverless GPU

专为运行机器学习工作负载打造的服务器less GPU云平台。当您需要无需管理基础设施即可按需使用GPU、将机器学习模型部署为API，或运行具备自动扩展功能的批处理作业时，可选用此平台。

技能元数据

参考：完整 SKILL.md 内容

:::info 以下是当触发该技能时Hermes所加载的完整技能定义。技能处于激活状态时，代理程序将依据此内容执行操作。

::

Modal Serverless GPU

关于在Modal的服务器less GPU云平台上运行机器学习工作负载的全面指南。

何时选择Modal

以下情况适合使用Modal：

• 在无需管理基础设施的情况下运行对GPU性能要求较高的机器学习工作负载
• 将机器学习模型部署为具备自动扩展功能的API
• 运行批处理作业（包括模型训练、推理及数据处理）
• 需要按使用时长计费的GPU资源，且无需承担空闲成本
• 快速构建机器学习应用原型
• 执行定时任务（类似cron的作业调度）

核心功能：

• 服务器less GPU：支持按需使用T4、L4、A10G、L40S、A100、H100、H200、B200等型号的GPU
• 原生Python支持：可通过Python代码定义基础设施，无需使用YAML格式
• 自动扩展：可瞬间将资源扩展至零，或扩展至100多台GPU
• 亚秒级冷启动：基于Rust技术的基础设施，可实现快速容器启动
• 容器缓存：对镜像层进行缓存，从而加快迭代速度
• Web端点：可将函数部署为REST API，并实现无中断更新

如需其他替代方案，请考虑：

• RunPod：适用于需要持久化状态的长时间运行的容器任务
• Lambda Labs：适用于预留GPU实例的场景
• SkyPilot：适用于多云环境下的任务编排及成本优化
• Kubernetes：适用于复杂的多服务架构

快速入门

安装

pip install modal
modal setup  # Opens browser for authentication

使用 GPU 实现“Hello World”示例

import modal

app = modal.App("hello-gpu")

@app.function(gpu="T4")
def gpu_info():
    import subprocess
    return subprocess.run(["nvidia-smi"], capture_output=True, text=True).stdout

@app.local_entrypoint()
def main():
    print(gpu_info.remote())

运行命令：modal run hello_gpu.py

基本推理接口

import modal

app = modal.App("text-generation")
image = modal.Image.debian_slim().pip_install("transformers", "torch", "accelerate")

@app.cls(gpu="A10G", image=image)
class TextGenerator:
    @modal.enter()
    def load_model(self):
        from transformers import pipeline
        self.pipe = pipeline("text-generation", model="gpt2", device=0)

    @modal.method()
    def generate(self, prompt: str) -> str:
        return self.pipe(prompt, max_length=100)[0]["generated_text"]

@app.local_entrypoint()
def main():
    print(TextGenerator().generate.remote("Hello, world"))

核心概念

主要组件

执行模式

GPU 配置

可用 GPU

GPU规格模式

# Single GPU
@app.function(gpu="A100")

# Specific memory variant
@app.function(gpu="A100-80GB")

# Multiple GPUs (up to 8)
@app.function(gpu="H100:4")

# GPU with fallbacks
@app.function(gpu=["H100", "A100", "L40S"])

# Any available GPU
@app.function(gpu="any")

容器镜像

# Basic image with pip
image = modal.Image.debian_slim(python_version="3.11").pip_install(
    "torch==2.1.0", "transformers==4.36.0", "accelerate"
)

# From CUDA base
image = modal.Image.from_registry(
    "nvidia/cuda:12.1.0-cudnn8-devel-ubuntu22.04",
    add_python="3.11"
).pip_install("torch", "transformers")

# With system packages
image = modal.Image.debian_slim().apt_install("git", "ffmpeg").pip_install("whisper")

持久化存储

volume = modal.Volume.from_name("model-cache", create_if_missing=True)

@app.function(gpu="A10G", volumes={"/models": volume})
def load_model():
    import os
    model_path = "/models/llama-7b"
    if not os.path.exists(model_path):
        model = download_model()
        model.save_pretrained(model_path)
        volume.commit()  # Persist changes
    return load_from_path(model_path)

Web 接口端点

FastAPI 接口端点装饰器

@app.function()
@modal.fastapi_endpoint(method="POST")
def predict(text: str) -> dict:
    return {"result": model.predict(text)}

完整的 ASGI 应用程序

from fastapi import FastAPI
web_app = FastAPI()

@web_app.post("/predict")
async def predict(text: str):
    return {"result": await model.predict.remote.aio(text)}

@app.function()
@modal.asgi_app()
def fastapi_app():
    return web_app

Web端点类型

动态批量处理

@app.function()
@modal.batched(max_batch_size=32, wait_ms=100)
async def batch_predict(inputs: list[str]) -> list[dict]:
    # Inputs automatically batched
    return model.batch_predict(inputs)

密钥管理

# Create secret
modal secret create huggingface HF_TOKEN=hf_xxx

@app.function(secrets=[modal.Secret.from_name("huggingface")])
def download_model():
    import os
    token = os.environ["HF_TOKEN"]

日程安排

@app.function(schedule=modal.Cron("0 0 * * *"))  # Daily midnight
def daily_job():
    pass

@app.function(schedule=modal.Period(hours=1))
def hourly_job():
    pass

性能优化

缓解冷启动问题

@app.function(
    container_idle_timeout=300,  # Keep warm 5 min
    allow_concurrent_inputs=10,  # Handle concurrent requests
)
def inference():
    pass

模型加载最佳实践

@app.cls(gpu="A100")
class Model:
    @modal.enter()  # Run once at container start
    def load(self):
        self.model = load_model()  # Load during warm-up

    @modal.method()
    def predict(self, x):
        return self.model(x)

并行处理

@app.function()
def process_item(item):
    return expensive_computation(item)

@app.function()
def run_parallel():
    items = list(range(1000))
    # Fan out to parallel containers
    results = list(process_item.map(items))
    return results

常见配置

@app.function(
    gpu="A100",
    memory=32768,              # 32GB RAM
    cpu=4,                     # 4 CPU cores
    timeout=3600,              # 1 hour max
    container_idle_timeout=120,# Keep warm 2 min
    retries=3,                 # Retry on failure
    concurrency_limit=10,      # Max concurrent containers
)
def my_function():
    pass

调试

# Test locally
if __name__ == "__main__":
    result = my_function.local()

# View logs
# modal app logs my-app

常见问题

参考资料

• 高级用法 - 多 GPU 使用、分布式训练、成本优化
• 故障排查 - 常见问题及解决方案

资源链接

• 文档：https://modal.com/docs
• 示例代码：https://github.com/modal-labs/modal-examples
• 定价信息：https://modal.com/pricing
• Discord 社群：https://discord.gg/modal