Skip to content

Commit ba5c84d

Browse files
committed
installing torch w cuda dependencies for non-windows and non-jetson envs
1 parent aad7554 commit ba5c84d

File tree

1 file changed

+1
-1
lines changed
  • qa/L0_backend_python/examples

1 file changed

+1
-1
lines changed

qa/L0_backend_python/examples/test.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@ pip3 uninstall -y torch
4141
pip3 uninstall -y numpy
4242
pip3 install "numpy>=2"
4343
if [ "$TEST_JETSON" == "0" ] && [[ ${TEST_WINDOWS} == 0 ]]; then
44-
pip3 install torch==2.5.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.20.0
44+
pip3 install torch==2.5.0 torchvision==0.20.0 --index-url https://download.pytorch.org/whl/cu124
4545
else
4646
pip3 install torch==2.5.0 -f https://download.pytorch.org/whl/torch_stable.html torchvision==0.20.0
4747
fi

0 commit comments

Comments
 (0)