Colab Pro: Zero-Touch Automation
Zero-Touch automated training on Google Colab Pro.
Features
- Automated GPU Allocation: Automatically selects and connects to the best available GPU (A100, V100, T4) based on your priority.
- Seamless Code Sync: Synchronizes your local code with Google Colab via Google Drive.
- One-Click Training: Start your training sessions directly from VS Code.
- Session Management: Monitor and stop Colab sessions without leaving your editor.
- Configurable: Easily configure notebook URLs, account details, and training paths.
Usage
- Open Configuration: Run the command
Colab Pro: Open Configuration to set up your notebook URL, account, and paths.
- Run Automation: Use
Colab Pro: Run Automation to start the automation.
- Stop Session: Use
Colab Pro: Stop Session to terminate the current session.
Configuration
The extension provides several settings:
colab-pro.notebookUrl: The URL of your Google Colab notebook.colab-pro.gpuPriority: Priority list for GPU allocation (e.g., ["A100", "V100", "T4"]).colab-pro.driveSyncPath: Local path to your Google Drive sync folder.colab-pro.account: Your Google/Colab account email.colab-pro.trainingFilePath: Path to the training file (e.g., a .py or .ipynb file).colab-pro.resultsSavePath: Path where training results should be saved.
Requirements
- A Google Colab Pro or Pro+ subscription.
- Google Drive Desktop installed and synced.
- Python installed locally for the automation engine.
License
MIT
| |
