From 232d4320932a6dc2fd05f824dbab6606de07adf8 Mon Sep 17 00:00:00 2001
From: GuanLuo <41310872+GuanLuo@users.noreply.github.com>
Date: Tue, 6 Dec 2022 11:34:59 -0800
Subject: [PATCH] Upgrade Pytorch version installed in tests (#5140)

---
 qa/L0_backend_python/bls/test.sh      | 2 +-
 qa/L0_backend_python/examples/test.sh | 2 +-
 qa/L0_backend_python/test.sh          | 2 +-
 qa/L0_backend_python/unittest/test.sh | 2 +-
 qa/L0_infer/test.sh                   | 4 ++--
 qa/L0_io/test.sh                      | 2 +-
 6 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/qa/L0_backend_python/bls/test.sh b/qa/L0_backend_python/bls/test.sh
index 62a98dd228..ec93267474 100644
--- 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
index 1d5a79df17..393540aafd 100644
--- 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
index 2667f48389..2cc92f913f 100755
--- 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
index e78b2613b0..a99d708e3c 100644
--- 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
index c2361185e2..038a7f2bae 100755
--- 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
index ac1ad5559e..b8830d9ee4 100755
--- 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