pytorch
7dfa7554 - Print only the driver version from the first GPU (#88364)

Commit
2 years ago
Print only the driver version from the first GPU (#88364) For example, distributed test has more than one of them: ``` nvidia-smi --query-gpu=driver_version --format=csv,noheader 515.57 515.57 ``` while `--id=0` correctly prints: ``` nvidia-smi --query-gpu=driver_version --format=csv,noheader --id=0 515.57 ``` This is to avoid re-install the same driver as in https://github.com/pytorch/pytorch/actions/runs/3380662072/jobs/5613981088 Pull Request resolved: https://github.com/pytorch/pytorch/pull/88364 Approved by: https://github.com/seemethere, https://github.com/ZainRizvi
Author
Committer
Parents
Loading