pg_gpu_info to get NVIDIA-related information like with nvidia-smi
#5
Conversation
jsaied99
changed the title
pg_gpu_info to get NVIDIA-related information like with nvidia-smi
|
Good start! Do you think it could be possible to make those metrics part of the Also curious, how fast is it to query this as compared to our |
|
|
||
| ```sql | ||
| ---- | ||
| SELECT device_id, device_name, total_memory_mb, used_memory_mb, temperature_c from pg_gpu_info(); |
There was a problem hiding this comment.
We want to namespace this function to this extension as well, if we are to use this approach. Like we do with PostgresML, it should be something like SELECT * FROM pg_stat_sysinfo.gpu_metrics() or similar.
There was a problem hiding this comment.
Hey @levkk! I will look at the performance of getting the gpu_metrics vs. the others. Namespacing the function is also a good idea.
There was a problem hiding this comment.
We should also update the pgrx dependencies. I didn't right off the bat because I was getting some errors, but we definitely should.
There was a problem hiding this comment.
Ah yeah, good call. Would be great to match the pgrx version here with what we have in postgresml/postgresml.
Adding
pg_gpu_infoto get NVIDIA-related information like withnvidia-smi