benchmark

Install the right numpy version matching PyTorch CI

#2640

Closed