search

pytorch / benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.
BSD 3-Clause "New" or "Revised" License
854 stars 275 forks source link

Pin yolo dep #2307

Closed eellison closed 3 months ago

eellison commented 3 months ago

This is the cause of https://github.com/pytorch/pytorch/issues/128845. It is upgrading to numpy 2.0 which has BC incompat changes.

xuzhao9 commented 3 months ago

@seemethere I am wondering if we should also change numpy version here? https://github.com/pytorch/builder/blob/3b032fc9e0b7a17b74eb421da52b264089e0a15e/wheel/build_wheel.sh#L144

We should keep the consistent numpy version in https://github.com/pytorch/benchmark/blob/main/requirements.txt#L20

xuzhao9 commented 3 months ago

Replaced by https://github.com/pytorch/benchmark/pull/2311