move pynvml install to setup.py (#5840)

Commit

1 year ago

move pynvml install to setup.py (#5840) Only install pynvml on nvidia gpus; not all accelerators