Fixed the metric so it works with ROCM librariers.

Megvii-BaseDetection / YOLOX

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

Apache License 2.0

9.15k stars 2.15k forks source link

Fixed the metric so it works with ROCM librariers. #1759

Open nlbutts opened 4 months ago

nlbutts commented 4 months ago

The metric file calls nvidia-smi directly. This change calls nvidia-smi and if that fails called rocm-smi.

CLAassistant commented 4 months ago

All committers have signed the CLA.