fix pynvml import failure

[lxning]

Description

Please read our CONTRIBUTING.md prior to creating your first pull request.

Please include a summary of the feature or issue being fixed. Please also include relevant motivation and context. List any dependencies that are required for this change.

Fixes #(issue)
#1881

Type of change

Please delete options that are not relevant.

• Bug fix (non-breaking change which fixes an issue)
• Breaking change (fix or feature that would cause existing functionality to not work as expected)
• New feature (non-breaking change which adds functionality)
• This change requires a documentation update

Feature/Issue validation/testing

Please describe the Unit or Integration tests that you ran to verify your changes and relevant result summary. Provide instructions so it can be reproduced.
Please also list any relevant details for your test configuration.

Only "import pynvml" in gpu host.

• Test A
  Logs for Test A

• Test B
  Logs for Test B

Checklist:

• Did you have fun?
• Have you added tests that prove your fix is effective or that this feature works?
• Has code been commented, particularly in hard-to-understand areas?
• Have you made corresponding changes to the documentation?

[rohithkrn]

@lxning curious why this was not an issue always and only showed up recently?

[agunapal]

Looks good. Shouldn't the CPU benchmark CI always be failing because of this?

[codecov]

Codecov Report

Merging #1882 (8df0285) into master (b30c341) will decrease coverage by 0.03%.
The diff coverage is 0.00%.

❗ Current head 8df0285 differs from pull request most recent head 01e41ab. Consider uploading reports for the commit 01e41ab to get more accurate results

@@            Coverage Diff             @@
##           master    #1882      +/-   ##
==========================================
- Coverage   44.99%   44.95%   -0.04%     
==========================================
  Files          63       63              
  Lines        2609     2609              
  Branches       60       60              
==========================================
- Hits         1174     1173       -1     
- Misses       1435     1436       +1     

Impacted Files	Coverage Δ
ts/metrics/system_metrics.py	28.81% <0.00%> (-1.70%)	⬇️

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

[lxning]

@lxning curious why this was not an issue always and only showed up recently?

@rohithkrn This issue happened recently due to PR. It causes CPU usage report different from previous.

[lxning]

Looks good. Shouldn't the CPU benchmark CI always be failing because of this?

@ankithagunapal I found PR caused cpu usage abnormal, but not sure if it is also related to recent CPU benchmark unstable.