OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

[BE] Move numba pinned version to requirements-ci.txt

Browse Source 
Specify different pinned versions for older vs newer python runtime

Pull Request resolved: https://github.com/pytorch/pytorch/pull/77476

Approved by: https://github.com/mehtanirav
This commit is contained in:
Nikita Shulga 2022-05-16 17:24:33 -07:00 committed by PyTorch MergeBot
parent 1f8049566f
commit 4596ecb4d2
2 changed files with 3 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
.circleci/docker/common/install_conda.sh
View File

@ -100,14 +100,6 @@ if [ -n "$ANACONDA_PYTHON_VERSION" ]; then
# Install some other packages, including those needed for Python test reporting
pip_install -r /opt/conda/requirements-ci.txt
# Install numba only on python-3.8 or below
# For numba issue see https://github.com/pytorch/pytorch/issues/51511
if [[ $(python -c "import sys; print(int(sys.version_info < (3, 9)))") == "1" ]]; then
pip_install numba==0.54.1
else
pip_install numba==0.49.0
fi
# Update scikit-learn to a python-3.8 compatible version
if [[ $(python -c "import sys; print(int(sys.version_info >= (3, 8)))") == "1" ]]; then
pip_install -U scikit-learn

4
.circleci/docker/requirements-ci.txt
View File

@ -98,10 +98,12 @@ mypy==0.812
#Pinned versions: 1.10.0.post1
#test that import: run_test.py, test_cpp_extensions_aot.py,test_determination.py
#numba
numba==0.49.0 ; python_version < "3.9"
numba==0.54.1 ; python_version == "3.9"
#Description: Just-In-Time Compiler for Numerical Functions
#Pinned versions: 0.54.1, 0.49.0, <=0.49.1
#test that import: test_numba_integration.py
#For numba issue see https://github.com/pytorch/pytorch/issues/51511
#numpy
#Description: Provides N-dimensional arrays and linear algebra