File tree Expand file tree Collapse file tree 1 file changed +14
-3
lines changed
qa/L0_backend_python/examples Expand file tree Collapse file tree 1 file changed +14
-3
lines changed Original file line number Diff line number Diff line change @@ -41,11 +41,22 @@ pip3 uninstall -y torch
4141pip3 uninstall -y numpy
4242# NOTE: Using this subtest as a test case that involves using a python model with
4343# numpy 2.X without changing the environments used in all the other test cases.
44- pip3 install " numpy>=2"
4544if [ " $TEST_JETSON " == " 0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
46- pip3 install torch==2.5.0 torchvision==0.20.0 --index-url https://download.pytorch.org/whl/cu124
45+ if [ ${PYTHON_ENV_VERSION} == " 8" ]; then
46+ pip3 install " numpy<2"
47+ pip3 install torch==2.0.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0+cu117
48+ else
49+ pip3 install " numpy>=2"
50+ pip3 install torch==2.5.0 torchvision==0.20.0 --index-url https://download.pytorch.org/whl/cu124
51+ fi
4752else
48- pip3 install torch==2.5.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.20.0
53+ if [ ${PYTHON_ENV_VERSION} == " 8" ]; then
54+ pip3 install " numpy<2"
55+ pip3 install torch==2.0.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0
56+ else
57+ pip3 install " numpy>=2"
58+ pip3 install torch==2.5.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.20.0
59+ fi
4960fi
5061
5162# Install `validators` for Model Instance Kind example
You can’t perform that action at this time.
0 commit comments