GPU Environment Management for Visual Studio Code

The genv extension lets you interactively control, configure and monitor the GPU resources that your Visual Studio Code session is using.

With genv you could:

• Easily share GPUs with your teammates.
• Find available GPUs for you to use.
• Switch between GPUs without code changes.

Getting Started

Read the genv reference to get started.

Installation

From the Visual Studio Marketplace.

Features

Setting the Device Count

You can configure how many GPUs you need just by clicking the status bar item. After specifying the device count, genv will look for available GPUs and attach them for your environment.

From now on, anything you'll run in your Visual Studio Code session would use only these GPUs (e.g. Python scripts, terminal commands, etc.). You will also see the devices turn green which shows that they are attached to your environment.

Choosing Specific GPUs to Use

In addition to automatically provisioning GPUs based on the configured device count, genv lets you choose specific GPUs that you want your environment to use. You can switch between GPUs easily by selecting them in the device view.

Development

Publish

Bump the version. You can use npm-version. For example:

npm version patch

Package the extension using:

npm run package

Publish the extension using:

npm run publish