File tree Expand file tree Collapse file tree 1 file changed +6
-3
lines changed
qa/L0_backend_python/examples Expand file tree Collapse file tree 1 file changed +6
-3
lines changed Original file line number Diff line number Diff line change @@ -29,7 +29,6 @@ source ../common.sh
2929source ../../common/util.sh
3030
3131TRITON_REPO_ORGANIZATION=${TRITON_REPO_ORGANIZATION:= " http://github.com/triton-inference-server" }
32-
3332SERVER_ARGS=" --model-repository=${MODELDIR} /examples/python_backend/models --backend-directory=${BACKEND_DIR} --log-verbose=1"
3433SERVER_LOG=" ./examples_server.log"
3534
@@ -38,10 +37,12 @@ rm -fr *.log python_backend/
3837
3938# Install torch
4039pip3 uninstall -y torch
40+ pip3 uninstall -y numpy
41+ pip3 install " numpy>=2"
4142if [ " $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
4344else
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
4546fi
4647
4748# Install `validators` for Model Instance Kind example
440441 echo -e " \n***\n*** Example verification test FAILED.\n***"
441442fi
442443
444+ pip3 uninstall numpy
445+ pip3 install " numpy<2"
443446exit $RET
You can’t perform that action at this time.
0 commit comments