diff --git a/qa/L0_python_client_unit_tests/test.sh b/qa/L0_python_client_unit_tests/test.sh index 20b30b51da..5a46ecccc5 100755 --- a/qa/L0_python_client_unit_tests/test.sh +++ b/qa/L0_python_client_unit_tests/test.sh @@ -29,6 +29,9 @@ TEST_LOG="./python_client_unit_tests.log" PYTHON_CLIENT_UNIT_TESTS_DIR=/opt/tritonserver/qa/python_client_unit_tests/ PYTHON_CLIENT_UNIT_TESTS_CMD="python3 -m unittest discover -v -s $PYTHON_CLIENT_UNIT_TESTS_DIR -t $PYTHON_CLIENT_UNIT_TESTS_DIR" +# DLPack test requires Torch to validate GPU tensor +pip3 install torch + RET=0 rm -f $TEST_LOG