Skip to content

Commit

Permalink
Installing lower version for python 3.8
Browse files Browse the repository at this point in the history
  • Loading branch information
KrishnanPrash committed Nov 1, 2024
1 parent ec3bccd commit 00fe74a
Showing 1 changed file with 14 additions and 3 deletions.
17 changes: 14 additions & 3 deletions qa/L0_backend_python/examples/test.sh
Original file line number Diff line number Diff line change
Expand Up @@ -41,11 +41,22 @@ pip3 uninstall -y torch
pip3 uninstall -y numpy
# NOTE: Using this subtest as a test case that involves using a python model with
# numpy 2.X without changing the environments used in all the other test cases.
pip3 install "numpy>=2"
if [ "$TEST_JETSON" == "0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
pip3 install torch==2.5.0 torchvision==0.20.0 --index-url https://download.pytorch.org/whl/cu124
if [ ${PYTHON_ENV_VERSION} == "8" ]; then
pip3 install "numpy<2"
pip3 install torch==2.0.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0+cu117
else
pip3 install "numpy>=2"
pip3 install torch==2.5.0 torchvision==0.20.0 --index-url https://download.pytorch.org/whl/cu124
fi
else
pip3 install torch==2.5.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.20.0
if [ ${PYTHON_ENV_VERSION} == "8" ]; then
pip3 install "numpy<2"
pip3 install torch==2.0.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.15.0
else
pip3 install "numpy>=2"
pip3 install torch==2.5.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.20.0
fi
fi

# Install `validators` for Model Instance Kind example
Expand Down

0 comments on commit 00fe74a

Please sign in to comment.