MCP Server Colab Exec

🚀 mcp-server-colab-exec

mcp-server-colab-exec 是一個 MCP 服務器，它能夠分配 Google Colab 的 GPU 運行時（T4/L4），並在其上執行 Python 代碼。藉助該服務器，任何與 MCP 兼容的 AI 助手，如 Claude Code、Claude Desktop、Gemini CLI、Cline 等，都可以在無需本地 GPU 硬件的情況下運行 GPU 加速代碼（CUDA、PyTorch、TensorFlow）。

🚀 快速開始

在使用 mcp-server-colab-exec 之前，你需要滿足以下前提條件：

• Python 3.10 及以上版本
• 擁有一個可以訪問 Google Colab 的 Google 賬戶
• 首次運行時，會打開一個瀏覽器窗口進行 OAuth2 授權。授權令牌會被緩存到 ~/.config/colab-exec/token.json，後續運行將使用該緩存令牌。

📦 安裝指南

你可以使用以下命令安裝 mcp-server-colab-exec：

pip install mcp-server-colab-exec

或者使用 uvx 直接運行：

uvx mcp-server-colab-exec

🛠️ 配置說明

Claude Code

你可以將以下配置添加到項目的 .mcp.json 或 ~/.claude/.mcp.json 文件中：

{
  "mcpServers": {
    "colab-exec": {
      "command": "mcp-server-colab-exec"
    }
  }
}

也可以通過 CLI 進行配置：

claude mcp add colab-exec mcp-server-colab-exec

Claude Desktop

將以下配置添加到 claude_desktop_config.json 文件中：

{
  "mcpServers": {
    "colab-exec": {
      "command": "mcp-server-colab-exec"
    }
  }
}

Gemini CLI

使用以下命令進行配置：

gemini mcp add colab-exec -- mcp-server-colab-exec

💻 使用示例

基礎用法

檢查 GPU 可用性

colab_execute(code="import torch; print(torch.cuda.is_available()); print(torch.cuda.get_device_name(0))")

運行 nvidia-smi

colab_execute(code="import subprocess; print(subprocess.run(['nvidia-smi'], capture_output=True, text=True).stdout)")

訓練模型並下載權重

colab_execute_notebook(
    code="import torch; model = torch.nn.Linear(10, 1); torch.save(model.state_dict(), '/tmp/model.pt')",
    output_dir="./outputs"
)

📚 詳細文檔

colab_execute

該工具用於在 Colab GPU 運行時執行內聯 Python 代碼。

colab_execute_file

該工具用於在 Colab GPU 運行時執行本地 .py 文件。

colab_execute_notebook

該工具用於執行代碼並收集所有生成的工件（如圖像、CSV、模型等）。

🔑 認證說明

首次使用時，服務器會打開一個瀏覽器窗口進行 Google OAuth2 授權。訪問令牌和刷新令牌會被緩存到 ~/.config/colab-exec/token.json。後續運行將使用緩存的令牌，並自動刷新。 OAuth2 客戶端憑證與官方 Google Colab VS Code 擴展（google.colab@0.3.0）使用的憑證相同，這些憑證是公開的。

❓ 故障排除

• “GPU 配額已超出” — Colab 有使用限制，請等待一段時間後重試，或者使用不同的 Google 賬戶。
• “創建內核會話超時” — 運行時啟動時間過長，請重試，因為 Colab 在高峰時段有時會出現延遲。
• “認證失敗” — 刪除 ~/.config/colab-exec/token.json 文件，然後重新進行認證。
• OAuth 瀏覽器窗口未打開 — 請確保你在支持瀏覽器的環境中運行。對於無頭服務器，可先在有瀏覽器的機器上進行認證，然後複製令牌文件。

📄 許可證

本項目採用 MIT 許可證。