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:
Read the genv reference to get started.
From the Visual Studio Marketplace.
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.
Bump the version.
You can use
Package the extension using:
Publish the extension using: