SuperAPI AI 工具集

agentic-ecology-colab-cli

智能体与模型

查看项目

更新于 10-15 21:53

技能介绍

用途:该技能提供在远程 Google Colab 运行时上启动、管理和执行工作负载的指南与工作流,用于把模型训练、数据集嵌入生成等重计算任务卸载到带 GPU/TPU 加速器的远程 Google Colab 虚拟机。

主要能力与流程:

  1. 验证 CLI 安装:确认环境中存在 Colab CLI,缺失时安装。
  2. 授权 CLI:检查活动会话验证凭据;若未认证,获取 CLI 生成的 OAuth 链接,展示给用户并等待其提供授权码完成登录。
  3. 创建远程会话:用唯一名称创建新远程会话,需要时选择 GPU 加速器,否则默认 CPU。
  4. 挂载 Google Drive:为该会话触发 Drive 挂载命令;若会话为无头/远程模式,需监控日志中的 OAuth URL,展示给用户并等待其授权。
  5. 安装依赖:在远程 VM 上安装所需的 Python 库或 requirements.txt 包。
  6. 执行处理脚本:将本地工作区的 Python 脚本执行到远程 VM 上。
  7. 清理 VM 资源:执行完成后立即停止会话,释放资源并避免计算单元费用。

使用方式:按上述顺序逐步执行,具体命令选项、认证模式和脚本执行命令结构见 Colab CLI 技术参考。

适用场景:需要 GPU/TPU 的模型训练、数据集嵌入生成等重计算任务卸载。

限制:依赖外部 google-colab-cli;需要用户交互完成 OAuth 授权;无头会话需监控日志;执行后应停止会话以控制费用。