Upgrade Pytorch version installed in tests (triton-inference-server#5140

)
shixiaoyu321 · Dec 6, 2022 · 232d432 · 232d432
1 parent fa18c45
commit 232d432
Show file tree

Hide file tree

Showing 6 changed files with 7 additions and 7 deletions.
diff --git a/qa/L0_backend_python/bls/test.sh b/qa/L0_backend_python/bls/test.sh
@@ -41,7 +41,7 @@ RET=0
 rm -fr *.log ./models
 
 pip3 uninstall -y torch
-pip3 install torch==1.11.0+cu113 -f https://download.pytorch.org/whl/torch_stable.html
+pip3 install torch==1.13.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html
 
 mkdir -p models/bls/1/
 cp ../../python_models/bls/model.py models/bls/1/

diff --git a/qa/L0_backend_python/examples/test.sh b/qa/L0_backend_python/examples/test.sh
@@ -41,7 +41,7 @@ rm -fr *.log python_backend/
 # Skip torch install on Jetson since it is already installed.
 if [ "$TEST_JETSON" == "0" ]; then
     pip3 uninstall -y torch
-    pip3 install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
+    pip3 install torch==1.13.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html
 fi
 
 # Install JAX

diff --git a/qa/L0_backend_python/test.sh b/qa/L0_backend_python/test.sh
@@ -130,7 +130,7 @@ cp ../python_models/string_fixed/config.pbtxt ./models/string_fixed
 
 # Skip torch install on Jetson since it is already installed.
 if [ "$TEST_JETSON" == "0" ]; then
-  pip3 install torch==1.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
+  pip3 install torch==1.13.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
 else
   # test_growth_error is skipped on jetson
   EXPECTED_NUM_TESTS=8

diff --git a/qa/L0_backend_python/unittest/test.sh b/qa/L0_backend_python/unittest/test.sh
@@ -40,7 +40,7 @@ source ../../common/util.sh
 
 # Uninstall the non CUDA version of PyTorch
 pip3 uninstall -y torch
-pip3 install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
+pip3 install torch==1.13.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html
 pip3 install tensorflow
 
 rm -fr *.log ./models

diff --git a/qa/L0_infer/test.sh b/qa/L0_infer/test.sh
@@ -137,9 +137,9 @@ export ENSEMBLES
 if [[ $BACKENDS == *"python_dlpack"* ]]; then
     if [ "$TEST_JETSON" == "0" ]; then
         if [[ "aarch64" != $(uname -m) ]] ; then
-            pip3 install torch==1.9.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
+            pip3 install torch==1.13.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
         else
-            pip3 install torch==1.9.0 -f https://download.pytorch.org/whl/torch_stable.html
+            pip3 install torch==1.13.0 -f https://download.pytorch.org/whl/torch_stable.html
         fi
     fi
 fi

diff --git a/qa/L0_io/test.sh b/qa/L0_io/test.sh
@@ -56,7 +56,7 @@ LD_LIBRARY_PATH=/opt/tritonserver/lib:$LD_LIBRARY_PATH
 rm -f $CLIENT_LOG.*
 
 # PyTorch is required for the Python backend dlpack add sub models
-pip3 install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html
+pip3 install torch==1.13.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html
 RET=0
 
 # Prepare float32 models with basic config