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
41
41
pip3 uninstall -y numpy
42
42
# NOTE: Using this subtest as a test case that involves using a python model with
43
43
# numpy 2.X without changing the environments used in all the other test cases.
44
- pip3 install " numpy>=2"
45
44
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
47
52
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
49
60
fi
50
61
51
62
# Install `validators` for Model Instance Kind example
You can’t perform that action at this time.
0 commit comments