Pin numba to 0.54.1 (#71327)
Summary:
Not sure what is going on, but numba=0.55.0 currently installed in for example 308535385114.dkr.ecr.us-east-1.amazonaws.com/pytorch/pytorch-linux-bionic-py3.7-clang9:0d18ad2827487386d2a7864b11fec5bc83de6545 is build against newer version of numpy, which was apparently silently fixed on the pypi side (as latest numba download is numba-0.55.0-1-cp37-cp37m-manylinux2014_x86_64.manylinux_2_17_x86_64.whl )
Fixes https://github.com/pytorch/pytorch/issues/71320
Pull Request resolved: https://github.com/pytorch/pytorch/pull/71327
Reviewed By: suo, seemethere, atalman, janeyx99
Differential Revision: D33589002
Pulled By: malfet
fbshipit-source-id: d362a2b2fd045bc1720cd7fdc4c7b18b7d607fc4