Elgato_dark/ultralytics
1
0
Fork 0
mirror of https://github.com/ultralytics/ultralytics synced 2026-04-21 22:17:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 8
ultralytics/docs/en/reference/utils/autodevice.md
Onuralp SEZER edd95b1dee
Replace pynvml with NVIDIA official nvidia-ml-py package (#21795) 
Signed-off-by: Onuralp SEZER <onuralp@ultralytics.com>
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com>
Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com>
2025-08-25 00:56:56 +02:00

948 B
Raw Blame History

description keywords
Discover how to automatically select the most idle GPU using PyTorch and NVIDIA (nvidia-ml-py) with this AutoDevice script. Ideal for optimizing GPU usage in deep learning and computer vision tasks. Ultralytics, AutoDevice, GPU selection, PyTorch, NVML, pynvml, GPU monitoring, CUDA, deep learning, idle GPU, memory utilization, temperature, power usage

Reference for ultralytics/utils/autodevice.py

!!! note

This file is available at [https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/autodevice.py](https://github.com/ultralytics/ultralytics/blob/main/ultralytics/utils/autodevice.py). If you spot a problem please help fix it by [contributing](https://docs.ultralytics.com/help/contributing/) a [Pull Request](https://github.com/ultralytics/ultralytics/edit/main/ultralytics/utils/autodevice.py) 🛠️. Thank you 🙏!

::: ultralytics.utils.autodevice.GPUInfo