Skip to content

ci: Adding tests for numpy>=2 #7756

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 11 commits into from
Nov 1, 2024
Merged

ci: Adding tests for numpy>=2 #7756

Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
70e08bc
Modify numpy, torch, torchvision versions for examples subtest
KrishnanPrash Oct 30, 2024
fb3e0f8
spacing
KrishnanPrash Oct 30, 2024
aa3059a
spacing
KrishnanPrash Oct 30, 2024
aad7554
removing unecessary numpy uninstall + reinstall lower version
KrishnanPrash Oct 31, 2024
ba5c84d
installing torch w cuda dependencies for non-windows and non-jetson envs
KrishnanPrash Oct 31, 2024
598fade
Merge branch 'main' into kprashanth-numpy2-test
KrishnanPrash Oct 31, 2024
ba904a2
Update qa/L0_backend_python/examples/test.sh
KrishnanPrash Nov 1, 2024
ec3bccd
formatting
KrishnanPrash Nov 1, 2024
00fe74a
Installing lower version for python 3.8
KrishnanPrash Nov 1, 2024
9fc4c55
comments
KrishnanPrash Nov 1, 2024
b58ff56
Merge branch 'main' into kprashanth-numpy2-test
KrishnanPrash Nov 1, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
23 changes: 21 additions & 2 deletions qa/L0_backend_python/examples/test.sh
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -38,10 +38,29 @@ rm -fr *.log python_backend/

# Install torch
pip3 uninstall -y torch
pip3 uninstall -y numpy
# NOTE: Using this subtest as a test case that involves using a python model with
# numpy 2.X without changing the environments used in all the other test cases.
if [ "$TEST_JETSON" == "0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
pip3 install torch==2.0.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0+cu117
if [ ${PYTHON_ENV_VERSION} == "8" ]; then
# Python 3.8 does not support numpy 2.x, so installing numpy1.x
pip3 install "numpy<2"
pip3 install torch==2.0.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0+cu117
else
# Python 3.9 >= supports numpy 2.x.
pip3 install "numpy>=2"
pip3 install torch==2.5.0 torchvision==0.20.0 --index-url https://download.pytorch.org/whl/cu124
fi
else
pip3 install torch==2.0.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0
if [ ${PYTHON_ENV_VERSION} == "8" ]; then
# Python 3.8 does not support numpy 2.x, so installing numpy1.x
pip3 install "numpy<2"
pip3 install torch==2.0.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0
else
# Python 3.9 >= supports numpy 2.x.
pip3 install "numpy>=2"
pip3 install torch==2.5.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.20.0
fi
fi

# Install `validators` for Model Instance Kind example
Expand Down
Loading