watermark icon indicating copy to clipboard operation
watermark copied to clipboard

Would love the option to get the GPU and Nvidia revision info

Open dartdog opened this issue 8 years ago • 3 comments

As getting the GPU and Nvidia revisions can be kind of problematic.. Appreciate this great addition to the toolbox..

dartdog avatar Jun 25 '17 13:06 dartdog

That's a good suggestion! A parameter like --gpu_info could be a nice addition. I am wondering though how to do this in a general way (e.g., some way that works across OS's and would also work with ATI cards).

For Nvidia cards, for example, the best way would probably be to use

!cat nvidia-smi

or

!cat /proc/driver/nvidia/gpus/0/information

In IPython/Jupyter Notebook

I will leave this issue open and hope someone has a good idea how to do this elegantly within watermark :)

rasbt avatar Jun 25 '17 16:06 rasbt

Yeh but I don't know of anyone using amd for mach learning? but appreciate the !command info! that is fine for now, I should know but,,. heck

dartdog avatar Jun 27 '17 00:06 dartdog

Not sure if this is still of interest, but I took a stab at adding an argument that would at least work for NVIDIA-based GPUs. Details in #90 will share more insight on what could be expected if merged.

907Resident avatar May 19 '23 21:05 907Resident