Skip to content

Commit 74e9978

Browse files
authored
tests: Update torch installation (#8598) (#8602)
1 parent a1ea854 commit 74e9978

4 files changed

Lines changed: 8 additions & 8 deletions

File tree

qa/L0_dlpack_multi_gpu/test.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
# Copyright 2023-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
#
44
# Redistribution and use in source and binary forms, with or without
55
# modification, are permitted provided that the following conditions
@@ -41,7 +41,7 @@ source ../common/util.sh
4141

4242
# Uninstall the non CUDA version of PyTorch
4343
pip3 uninstall -y torch
44-
pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
44+
pip3 install torch -f https://download.pytorch.org/whl/cu130
4545

4646
# Install CuPy for testing non_blocking compute streams
4747
pip3 install cupy-cuda13x

qa/L0_io/test.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
# Copyright 2019-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright 2019-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
#
44
# Redistribution and use in source and binary forms, with or without
55
# modification, are permitted provided that the following conditions
@@ -55,7 +55,7 @@ LD_LIBRARY_PATH=/opt/tritonserver/lib:$LD_LIBRARY_PATH
5555
rm -f $CLIENT_LOG*
5656

5757
# PyTorch is required for the Python backend dlpack add sub models
58-
pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
58+
pip3 install torch -f https://download.pytorch.org/whl/cu130
5959
RET=0
6060

6161
# Prepare float32 models with basic config

qa/L0_libtorch_instance_group_kind_model/test.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
# Copyright 2023-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright 2023-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
#
44
# Redistribution and use in source and binary forms, with or without
55
# modification, are permitted provided that the following conditions
@@ -39,7 +39,7 @@ if [ ! -z "$TEST_REPO_ARCH" ]; then
3939
fi
4040

4141
pip3 uninstall -y torch
42-
pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
42+
pip3 install torch -f https://download.pytorch.org/whl/cu130
4343

4444
DATADIR=/data/inferenceserver/${REPO_VERSION}/qa_model_repository
4545
SERVER=/opt/tritonserver/bin/tritonserver

qa/L0_warmup/test.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,5 @@
11
#!/bin/bash
2-
# Copyright 2019-2025, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2+
# Copyright 2019-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
33
#
44
# Redistribution and use in source and binary forms, with or without
55
# modification, are permitted provided that the following conditions
@@ -420,7 +420,7 @@ wait $SERVER_PID
420420
# Test the onnx model to verify that the memory type of the output tensor
421421
# remains unchanged with the warmup setting
422422
pip3 uninstall -y torch
423-
pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
423+
pip3 install torch -f https://download.pytorch.org/whl/cu130
424424

425425
rm -fr models && mkdir models
426426
cp -r /data/inferenceserver/${REPO_VERSION}/qa_model_repository/onnx_nobatch_float32_float32_float32 models/.

0 commit comments

Comments
 (0)