技能：VAST.ai GPU 租赁

概述

此技能允许您按需配置 GPU 基础设施。在执行写入操作前，您必须拥有用户提供的 API 密钥。

功能

1. 搜索：根据 GPU 型号（例如 "RTX 4090"）和最高时价查找可用机器。
2. 租赁：在特定的 Offer ID 上实例化一个容器（默认使用 PyTorch 镜像）。
3. 连接：获取活跃实例的 SSH 连接字符串。
4. 余额：查询可用余额以及所有活跃机器当前的总时租消耗速率。

使用流程

• 步骤 1：如果上下文中没有，则向用户询问其 VAST API 密钥。
• 预检：在租赁前，调用 balance 功能以确保用户有足够资金。
• 步骤 2：搜索可用报价，并向用户展示最便宜的 3 个选项。
• 步骤 3：获得用户确认后，调用 rent 功能。
• 报告：如果余额低于 5.00 美元，则在每次成功租赁后警告用户。
• 步骤 4：等待 30-60 秒，然后调用 connect 功能以提供 SSH 连接字符串。

工具定义

• search(gpu: string, price: number)
• rent(id: number, image: string)
• connect(id: number)
• balance()

执行

运行这些工具的 CLI 封装程序。
命令：node /Users/sschepis/Development/vast-ai/dist/cli.js <action> [params]
环境变量：必须设置 VAST_API_KEY。

示例

• 搜索：node dist/cli.js search --gpu "RTX 4090" --price 0.5
• 租赁：node dist/cli.js rent --id 12345 --image "pytorch/pytorch"
• 连接：node dist/cli.js connect --id 12345
• 余额：node dist/cli.js balance