mcp服务器colab exec
MCP服务器，分配Google Colab GPU运行时（T4/L4）并在其上执行Python代码。允许任何兼容MCP的AI助手——Claude Code、Claude Desktop、Gemini CLI、Cline等——在没有本地GPU硬件的情况下运行GPU加速代码（CUDA、PyTorch、TensorFlow）。
先决条件
- Python 3.10+
- 有权访问的Google帐户 谷歌Colab
- 首次运行时，会打开一个浏览器窗口以获取OAuth2的同意。令牌缓存在
~/.config/colab-exec/token.json用于后续运行。
安装
pip install mcp-server-colab-exec或者直接与 uvx:
uvx mcp-server-colab-exec配置
克劳德代码
添加到您的项目 .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_config.json:
{
"mcpServers": {
"colab-exec": {
"command": "mcp-server-colab-exec"
}
}
}双子星命令行工具
gemini mcp add colab-exec -- mcp-server-colab-exec工具
colab_execute
在Colab GPU运行时上执行内联Python代码。
| 参数 | 类型 | 默认值 | 说明 |
|---|---|---|---|
code | string | -- | 要执行的Python代码（必需） |
accelerator | 字符串 | "T4" | GPU类型： "T4" （免费）或 "L4" （保费） |
timeout | int | 300 | 最大执行时间（秒） |
返回包含每个单元格输出、错误和stderr的JSON。
colab_execute_file
执行本地 .py Colab GPU运行时上的文件。
| 参数 | 类型 | 默认值 | 说明 |
|---|---|---|---|
file_path | string | -- | 本地路径 .py 文件（必填） |
accelerator | 字符串 | "T4" | GPU类型： "T4" （免费）或 "L4" （保费） |
timeout | int | 300 | 最大执行时间（秒） |
安全策略： file_path 必须是 .py 当前工作区内的文件(cwd).
colab_execute_notebook
执行代码并收集所有生成的工件（图像、CSV、模型等）。
| 参数 | 类型 | 默认值 | 说明 |
|---|---|---|---|
code | string | -- | 要执行的Python代码（必需） |
output_dir | string | -- | 下载工件的本地目录（必需） |
accelerator | 字符串 | "T4" | GPU类型： "T4" （免费）或 "L4" （保费） |
timeout | int | 300 | 最大执行时间（秒） |
工件以zip格式下载并提取到 output_dir. 在提取之前验证Zip成员，以防止路径遍历和特殊文件写入。
例子
检查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"
)认证
首次使用时，服务器会打开一个浏览器窗口，征求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浏览器窗口未打开 --确保您在有浏览器的环境中运行。对于无头服务器，首先使用浏览器在计算机上进行身份验证，然后复制令牌文件。
许可证
麻省理工学院