Google 推出 Colab CLI,讓開發者和 AI 代理在終端中遠程使用 Colab GPU 和 TPU 運行 Python
Google 發佈了 Colab CLI,這是一個命令行工具,允許開發者和 AI 代理在本地終端中連接遠程 Colab 運行時,在雲 GPU 和 TPU 上運行代碼。該工具開源(Apache 2.0 許可),支持創建會話、運行代碼、管理文件,並可與 Claude Code、Codex 和 Antigravity 等代理配合使用。
本週,Google AI 團隊發佈了 Colab CLI,這是一個將本地終端連接到遠程 Colab 運行時的命令行工具。它讓開發者和 AI 代理能夠在雲 GPU 和 TPU 上運行代碼,而全程無需離開終端。該工具在 Apache 2.0 許可下開源。
什麼是 Colab CLI?
Colab CLI 是 Google Colab 的命令行界面。用户可以在終端中創建會話、運行代碼和管理文件。任何具有終端訪問權限的代理都可以調用該工具,包括 Claude Code、Codex 和 Google 的 Antigravity。Google 提供了一個名為 COLAB_SKILL.md 的預打包技能文件,為代理提供如何使用 CLI 的上下文。
安裝只需一條 uv tool install 命令:
uv tool install git+https://github.com/googlecolab/google-colab-cli一個最小化會話如下:
colab new # 配置一個 CPU 會話
echo "print('hello')" | colab exec # 運行代碼
colab stop # 釋放虛擬機命令詳解
CLI 的命令分為會話、執行、文件和自動化。colab new 用於配置會話,默認使用 CPU。添加 --gpu T4、--gpu L4、--gpu A100 或 --gpu H100 可使用 GPU;TPU 選項為 v5e1 和 v6e1。colab exec 從標準輸入、.py 文件或筆記本運行 Python。它讀取本地文件並傳輸其內容,因此本地編輯無需單獨的上傳步驟。colab stop 終止會話並釋放虛擬機。
其他命令涵蓋文件和身份驗證。colab upload 和 colab download 在本地和遠程之間傳輸文件。colab drivemount 掛載 Google Drive,默認掛載到 /content/drive。colab auth 為 Google Cloud 服務驗證虛擬機。
核心循環:colab exec 與結果恢復
核心循環很簡短:配置運行時 → 運行腳本 → 拉回結果。colab download 檢索模型、數據集和其他文件。colab log 將會話歷史導出為 .ipynb、.md、.txt 或 .jsonl 格式。因此,遠程運行變成了磁盤上可重放的筆記本。colab repl 和 colab console 提供對虛擬機的交互式訪問。colab install 使用 uv 安裝軟件包,並回退到 pip。會話元數據存儲在 ~/.config/colab-cli/sessions.json。
示例:微調 Gemma 3 1B
Google 的官方發佈演示了一個代理驅動的微調任務。該任務使用 QLoRA 對 google/gemma-3-1b-it 進行微調,在 Text-to-SQL 數據集上訓練以改進 SQL 生成。Antigravity 代理通過五條命令運行完整的流水線:
colab new --gpu T4
colab install transformers datasets peft trl bitsandbytes accelerate
colab exec -f finetune_run.py
colab log --output gemma_finetune_log.ipynb
colab stop然後代理下載適配器模型、適配器配置、分詞器配置和分詞器。用户無需手動輸入雲配置命令。
用例
- 將筆記本上的訓練卸載到遠程 GPU 或 TPU,無需離開終端。
- 讓 Claude Code、Codex 或 Antigravity 等代理運行端到端 ML 流水線。
- 使用 QLoRA 遠程微調小型模型(如 Gemma 3 1B)。
- 將筆記本執行腳本化並導出可重放的 .ipynb 日誌,實現可重複性。
- 通過 colab repl 或 colab console 在虛擬機上交互式調試。
Colab CLI 與基於瀏覽器的 Colab 對比
CLI 並不取代筆記本 UI,而是針對腳本化、自動化和代理驅動的工作。以下是兩種工作流在常見任務上的對比:
| 維度 | 基於瀏覽器的 Colab | Colab CLI |
|------|-------------------|-----------|
| 界面 | Web 筆記本 UI | 本地終端 |
| 加速器選擇 | 瀏覽器中的運行時菜單 | colab new 的 --gpu/--tpu 標誌 |
| 代理使用 | 手動、UI 驅動 | 任何終端代理通過命令操作 |
| 運行本地腳本 | 粘貼或上傳到單元格 | colab exec -f script.py |
| 結果恢復 | 手動下載或 Drive | colab download、colab log |
| 軟件包安裝 | 單元格內的 !pip | colab install(uv,後 pip) |
| 會話控制 | 瀏覽器管理的運行時 | colab new、colab stop、colab status |
| 代理技能文件 | 無 | 內置 COLAB_SKILL.md |
優勢與考慮
優勢:
- 終端原生工作流,適合腳本、CI 和代理循環。
- 一條命令即可配置 T4、L4、A100 或 H100 GPU。
- exec 傳輸本地文件內容,無需上傳步驟。
- 日誌導出為可重放的筆記本格式,保證可重複性。
- 在 Apache 2.0 許可下開源,並附帶代理技能文件。
- 支持多種代理,不侷限於單一供應商。
考慮:
- 訪問需要身份驗證,默認策略為 oauth2。
- repl 和 console 在交互式運行時需要 TTY。
- 在腳本中使用這兩個命令需通過管道傳輸標準輸入。
- 計算仍在 Colab 的後端及其運行時模型上運行。
關鍵要點
- Google 的 Colab CLI 允許從本地終端在遠程 Colab GPU 和 TPU 上運行代碼。
- 一條命令配置加速器:
colab new --gpu T4至 A100 和 H100,以及 TPU。 colab exec將本地 .py 和 .ipynb 文件傳輸到運行時,無需上傳步驟。- 任何終端代理(Claude Code、Codex、Antigravity)均可通過內置的 COLAB_SKILL.md 驅動。
- 開源(Apache 2.0),
colab log導出可重放的筆記本日誌。