Skip to content

Commit ada98a7

Browse files
authored
Change torch versions (#7806)
1 parent 7c5be04 commit ada98a7

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

qa/L0_backend_python/decoupled/test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ SERVER_LOG="./decoupled_server.log"
3535
pip3 uninstall -y torch
3636
# FIXME: Until Windows supports GPU tensors, only test CPU scenarios
3737
if [[ ${TEST_WINDOWS} == 1 ]]; then
38-
pip3 install torch==1.13.0 -f https://download.pytorch.org/whl/torch_stable.html
38+
pip3 install torch==2.3.1 -f https://download.pytorch.org/whl/torch_stable.html
3939
else
4040
pip3 install torch==2.3.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html
4141
fi

qa/L0_backend_python/test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -172,7 +172,7 @@ if [ "$TEST_JETSON" == "0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
172172
pip3 install torch==2.3.1+cpu -f https://download.pytorch.org/whl/torch_stable.html
173173
else
174174
# GPU tensor tests are disabled on jetson
175-
pip3 install torch==1.13.0 -f https://download.pytorch.org/whl/torch_stable.html
175+
pip3 install torch==2.3.1 -f https://download.pytorch.org/whl/torch_stable.html
176176
fi
177177

178178
pip3 install pytest requests virtualenv

0 commit comments

Comments
 (0)