Skip to content

Commit 70e08bc

Browse files
committed
Modify numpy, torch, torchvision versions for examples subtest
1 parent afe4270 commit 70e08bc

File tree

1 file changed

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

1 file changed

+6
-3
lines changed

qa/L0_backend_python/examples/test.sh

Lines changed: 6 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,7 +29,6 @@ source ../common.sh
2929
source ../../common/util.sh
3030

3131
TRITON_REPO_ORGANIZATION=${TRITON_REPO_ORGANIZATION:="http://github.com/triton-inference-server"}
32-
3332
SERVER_ARGS="--model-repository=${MODELDIR}/examples/python_backend/models --backend-directory=${BACKEND_DIR} --log-verbose=1"
3433
SERVER_LOG="./examples_server.log"
3534

@@ -38,10 +37,12 @@ rm -fr *.log python_backend/
3837

3938
# Install torch
4039
pip3 uninstall -y torch
40+
pip3 uninstall -y numpy
41+
pip3 install "numpy>=2"
4142
if [ "$TEST_JETSON" == "0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
42-
pip3 install torch==2.0.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0+cu117
43+
pip3 install torch==2.5.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.20.0
4344
else
44-
pip3 install torch==2.0.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0
45+
pip3 install torch==2.5.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.20.0
4546
fi
4647

4748
# Install `validators` for Model Instance Kind example
@@ -440,4 +441,6 @@ else
440441
echo -e "\n***\n*** Example verification test FAILED.\n***"
441442
fi
442443

444+
pip3 uninstall numpy
445+
pip3 install "numpy<2"
443446
exit $RET

0 commit comments

Comments
 (0)