Skip to main content

Pods — @mariozechner/pi-pods

概述

pi 简化了在远程 GPU Pod 上运行大语言模型的过程,它自动:

  • 在新的 Ubuntu Pod 上设置 vLLM
  • 为 Agent 模型(Qwen、GPT-OSS、GLM 等)配置工具调用
  • 在同一 Pod 上管理多个模型,智能分配 GPU
  • 为每个模型提供 OpenAI 兼容的 API 端点
  • 包含用于测试的交互式 Agent(带文件系统工具)

安装

npm install -g @mariozechner/pi

快速开始

# 设置环境变量
export HF_TOKEN=your_huggingface_token
export PI_API_KEY=your_api_key

# 设置 DataCrunch Pod
pi pods setup dc1 "ssh root@1.2.3.4" \
--mount "sudo mount -t nfs ..."

# 启动模型
pi start Qwen/Qwen2.5-Coder-32B-Instruct --name qwen

# 发送单条消息
pi agent qwen "什么是斐波那契数列?"

# 交互式聊天模式
pi agent qwen -i

# 使用 OpenAI 兼容客户端
export OPENAI_BASE_URL='http://1.2.3.4:8001/v1'
export OPENAI_API_KEY=$PI_API_KEY

前置要求

  • Node.js 18+
  • HuggingFace Token
  • GPU Pod:
    • Ubuntu 22.04 或 24.04
    • SSH root 访问
    • NVIDIA 驱动已安装
    • 持久化存储(存放模型)

支持的 Provider

Provider说明
DataCrunch适合共享模型存储,NFS 卷可在同一区域的多个 Pod 间共享
即将支持Lambda Labs、Massed Compute、Paperspace、RunPod、Vast.ai

相关文档


翻译自 @mariozechner/pi-pods