Visual Studio Code Tools for AI
Visual Studio Code Tools for AI is an extension to build, test, and deploy machine learning and deep learning models. It seamlessly integrates with Azure Machine Learning for robust experimentation capabilities, including but not limited to running data preparation and model training experiments both locally and on remote compute targets.
Training a model using Azure Machine Learning
- Release notes
- Start page
- Prepare development environment
- Deep learning sample recipes
- View deep learning document in VS Code
- Frequently Asked Questions
Using Azure Machine Learning
- Getting started with Azure Machine Learning
- Create and manage Azure compute targets
- Train and tune models
- Deploy and manage models
- TensorFlow + Python
- Train models in the cloud
- Manage job history
- Manage storage
- Train models in PAI
- Open Jupyter notebooks in VS Code
- Run TensorBoard locally in VS Code
Supported Operating Systems
Currently this extension supports the following 64-bit operating systems:
Develop deep learning and AI solutions across Windows and MacOS
Because it's an IDE we've enabled familiar code editor features like syntax highlighting, IntelliSense (auto-completion) and text auto formatting. You can interactively test your deep learning application in your local environment using step-through debugging on local variables and models.
Get started quickly with the Start Page
Tools for AI Start Page is built to accelerate your start in AI world with
- Easy instructions to guide you to build your first AI application within 3 steps;
- AI inferencing/training samples and AI related learning materials for you to quickly learn and build your own AI solutions.
View deep learning document in VS Code
VS Code Tools for AI is shipped with document for some common deep learning frameworks. You can manage and browse the document in VS Code and search for API reference in editor window through simple command.
Find and share examples via the gallery
Visual Studio Code Tools for AI is integrated with Azure Machine Learning to make it easy to browse through a gallery of sample experiments using CNTK, TensorFlow, MMLSpark and more. This makes it easy to learn and share with others.
Scale out deep learning model training and/or inferencing to the cloud
This extension makes it easy to train models on your local computer or you can submit jobs to the cloud by using our integration with Azure Machine Learning. You can submit jobs to different compute targets like Spark clusters, Azure GPU virtual machines and more. Besides, Open Platform for AI (PAI) is also supported.
View recent job performance and details
Once the jobs are submitted, you can list the jobs, check the job details and download models, logs, etc. from the run history.
Manage storage on compute targets
The extension provides a storage explorer which enables you to manage your files on remote machines, PAI clusters, etc. You can create/delete folders, upload/download files within the VS Code easily.
Jupyter notebook server manager
The extension implements a built-in Jupyter server notebook server manager, which enables you to open a Jupyter notebook inside VS Code for viewing and editing.
Run TensorBoard locally in VS Code
The extension enables you to run TensorBoard in VS Code and view the model graph in an external browser. You could shutdown the TensorBoard server anytime in the VS Code terminal.
The extension provides several commands in the Command Palette for working with deep learning and machine learning:
- AI: Azure - Login: Login to Azure to access resources used by Azure ML.
- AI: Azure - Set Subscription: Set your Azure Subscription to use for Azure ML.
- AI: Open Azure ML Sample Explorer: Quickly get started with machine learning and deep learning experimentation by downloading sample projects you can run and modify to meet your needs.
- AI: Azure ML - Open Terminal: Open Azure ML CLI terminal to access full feature set.
- AI: Add Platform Configuration: Configure compute target (remote VM, PAI cluster) used for training job.
- AI: Edit Platform Configuration: Modify compute target settings.
- AI: Remove Platform Configuration: Remove a compute target configuration.
- AI: Submit Job: Submit a training job to remote Linux VMs, PAI clusters etc.
- AI: Edit Job Properties: Modify job settings.
- AI: List Jobs: View list of recent jobs you've submitted and their details.
- AI: Open Storage Explorer: View and manage storage on remote compute targets.
- AI: Local - Run TensorBoard: Run TensorBoard locally.
Support for this extension is provided on our GitHub Issue Tracker. You can submit a bug report, a feature suggestion or participate in discussions.
Code of Conduct
The Microsoft Enterprise and Developer Privacy Statement describes the privacy statement of this software.
This extension is subject to the terms of the End User License Agreement.