Skip to content

rshin/getgpu

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

8 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Synopsis

Return the ID of a GPU which has no compute processes running on it.

GPUs which have compute processes running on it according to NVML, and GPUs for which a different instance of getgpu has returned as available in the last --startup-time seconds (30 by default), are treated as occupied.

Usage

To install:

pip install --user git+https://github.com/rshin/getgpu

To set CUDA_VISIBLE_DEVICES:

getgpu -- [program to run] ...

Run getgpu --help to see options.

Development

pip install -e . dev
pre-commit install

About

Script to get available GPU ID.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages