reframe icon indicating copy to clipboard operation
reframe copied to clipboard

Auto-detect GPU devices

Open vkarak opened this issue 4 years ago • 2 comments

This is also mentioned in https://github.com/archspec/archspec/issues/25 but we could start working on a solution for ReFrame already.

vkarak avatar Nov 30 '21 16:11 vkarak

I was thinking about this as a part of #2276. As discussed in the archspec issue, we can use nvidia-smi to get a plethora of details about NVIDIA GPUs. For the AMD ones, there is an equivalent tool with python bindings. Besides detecting, if we could provide available memory, compute capabilities, threshold temperatures, etc., that would be quite useful!

mahendrapaipuri avatar Dec 01 '21 09:12 mahendrapaipuri

There is also this: https://pypi.org/project/nvidia-ml-py/

teojgo avatar Jan 11 '23 17:01 teojgo