Ultralytics:main
from
ultralytics:glenn-jocher-patch-1
description | keywords |
---|---|
Discover how to automatically select the most idle GPU using PyTorch and NVML (pynvml) 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 |
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 ๐!
Press p or to see the previous file or, n or to see the next file