A simple tool to select the optimal CUDA device based on memory, power, or utilization.
Project description
Auto Cuda Selector
A simple tool to select the optimal CUDA device based on memory, power, or utilization.
Install
pip install cuda-selector
Usage
from cuda_selector import auto_cuda
# Select cuda device with most memory available
device = auto_cuda()
# Select cuda device with lowest power usasge usage
device = auto_cuda('power')
# Select cuda device with lowest power usasge usage
device = auto_cuda('utilization')
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
cuda_selector-0.1.3.tar.gz
(2.6 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file cuda_selector-0.1.3.tar.gz.
File metadata
- Download URL: cuda_selector-0.1.3.tar.gz
- Upload date:
- Size: 2.6 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.11.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
5f9273f22d9d2d6a8128a3b69c1b72a904a52cbb1cab591b3f9e89608a258c76
|
|
| MD5 |
e9d5ef300ff5a9bab472febf551287be
|
|
| BLAKE2b-256 |
684f410ac5ff32e141def670e6eb2c0efc7cafb62b67a50c669044371f197d7c
|
File details
Details for the file cuda_selector-0.1.3-py3-none-any.whl.
File metadata
- Download URL: cuda_selector-0.1.3-py3-none-any.whl
- Upload date:
- Size: 2.7 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.11.10
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
bc44c7c34216d02af3df27eb85dedfe0f0dd699c1f7ed1b733a6e1d277269c7d
|
|
| MD5 |
7b81ed0e287df04d3070088d737cc34b
|
|
| BLAKE2b-256 |
486a3afcdd12795b945e9ecba90bfe427ea55f09a244bb78b8cc5dafc01b8c53
|
