Skip to content

Commit 00fe74a

Browse files
committed
Installing lower version for python 3.8
1 parent ec3bccd commit 00fe74a

File tree

1 file changed

+14
-3
lines changed
  • qa/L0_backend_python/examples

1 file changed

+14
-3
lines changed

qa/L0_backend_python/examples/test.sh

Lines changed: 14 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -41,11 +41,22 @@ pip3 uninstall -y torch
4141
pip3 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"
4544
if [ "$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
4752
else
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
4960
fi
5061

5162
# Install `validators` for Model Instance Kind example

0 commit comments

Comments
 (0)