xla
16a40cc6 - Fix get_gpu_name returning empty name. (#5925)

Commit
2 years ago
Fix get_gpu_name returning empty name. (#5925) Repro in python shell >>> subprocess.check_output(["nvidia-smi", "--query-gpu=gpu_name", "--format=csv"], encoding='utf-8').split("\n")[1:] ['NVIDIA A100-SXM4-40GB', ''] Tested fix >>> subprocess.check_output(["nvidia-smi", "--query-gpu=gpu_name", "--format=csv"], encoding='utf-8').split("\n")[1:-1] ['NVIDIA A100-SXM4-40GB']
Parents
Loading