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.
Get started with deep learning using PyTorch, TensorFlow, or other machine learning or deep-learning frameworks today.
Training a model using Azure Machine Learning
Getting Started
- Release notes
- Installation
- 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
Quickstarts
- 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
Currently this extension supports the following 64-bit operating systems:
- Windows
- macOS
VS Code Tools for AI is a cross-platform extension that supports deep learning frameworks including Microsoft Cognitive Toolkit (CNTK), Google TensorFlow and more.
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.
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.
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.
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.
Learn more about creating projects from the sample gallery
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.
Learn more about training models in the cloud
Once the jobs are submitted, you can list the jobs, check the job details and download models, logs, etc. from the run history.
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.
Learn more about storage explorer
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.
Learn more about Jupyter notebook 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.
This project has adopted the Microsoft Open Source Code of Conduct. For more information see the Code of Conduct FAQ or contact opencode@microsoft.com with any additional questions or comments.
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.