Fix torch installation

triton-inference-server · Nov 14, 2024 · 5ce86ab · 5ce86ab
1 parent d8eb00d
commit 5ce86ab
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/qa/L0_dlpack_multi_gpu/test.sh b/qa/L0_dlpack_multi_gpu/test.sh
@@ -41,7 +41,7 @@ source ../common/util.sh
 
 # Uninstall the non CUDA version of PyTorch
 pip3 uninstall -y torch
-pip3 install torch==1.13.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html
+pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
 pip3 install tensorflow
 
 # Install CuPy for testing non_blocking compute streams

diff --git a/qa/L0_infer/test.sh b/qa/L0_infer/test.sh
@@ -144,9 +144,9 @@ export BATCH
 
 if [[ $BACKENDS == *"python_dlpack"* ]]; then
     if [[ "aarch64" != $(uname -m) ]] ; then
-        pip3 install torch==2.2.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
+        pip3 install torch==2.3.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
     else
-        pip3 install torch==2.2.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
 fi
 

diff --git a/qa/L0_io/test.sh b/qa/L0_io/test.sh
@@ -1,5 +1,5 @@
 #!/bin/bash
-# Copyright 2019-2023, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# Copyright 2019-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
 # modification, are permitted provided that the following conditions
@@ -54,7 +54,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.13.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html
+pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
 RET=0
 
 # Prepare float32 models with basic config

diff --git a/qa/L0_warmup/test.sh b/qa/L0_warmup/test.sh
@@ -415,7 +415,7 @@ wait $SERVER_PID
 # Test the onnx model to verify that the memory type of the output tensor
 # remains unchanged with the warmup setting
 pip3 uninstall -y torch
-pip3 install torch==1.13.0+cu117 -f https://download.pytorch.org/whl/torch_stable.html
+pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
 
 rm -fr models && mkdir models
 cp -r /data/inferenceserver/${REPO_VERSION}/qa_model_repository/onnx_nobatch_float32_float32_float32 models/.