Change torch versions (#7806)

triton-inference-server · Nov 23, 2024 · be7a8cd · be7a8cd
1 parent 905a461
commit be7a8cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/qa/L0_backend_python/decoupled/test.sh b/qa/L0_backend_python/decoupled/test.sh
@@ -35,7 +35,7 @@ SERVER_LOG="./decoupled_server.log"
 pip3 uninstall -y torch
 # FIXME: Until Windows supports GPU tensors, only test CPU scenarios
 if [[ ${TEST_WINDOWS} == 1 ]]; then
-  pip3 install torch==1.13.0 -f https://download.pytorch.org/whl/torch_stable.html
+  pip3 install torch==2.3.1 -f https://download.pytorch.org/whl/torch_stable.html
 else
   pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
 fi

diff --git a/qa/L0_backend_python/test.sh b/qa/L0_backend_python/test.sh
@@ -172,7 +172,7 @@ if [ "$TEST_JETSON" == "0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
   pip3 install torch==2.3.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
 else
   # GPU tensor tests are disabled on jetson
-  pip3 install torch==1.13.0 -f https://download.pytorch.org/whl/torch_stable.html
+  pip3 install torch==2.3.1 -f https://download.pytorch.org/whl/torch_stable.html
 fi
 
 pip3 install pytest requests virtualenv