Skip to content

Commit 9cd265e

Browse files
committed
fix torchvision==0.20.1
Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com>
1 parent 95bac7c commit 9cd265e

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

.azure-pipelines/scripts/ut/3x/run_3x_pt.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ echo "##[section]import check pass"
1313
echo "##[group]set up UT env..."
1414
export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
1515
pip install -r /neural-compressor/test/3x/torch/requirements.txt
16-
pip install torch==2.5.1 # For auto-round
16+
pip install torch==2.5.1 torchvision==0.20.1 # For auto-round
1717
pip install pytest-cov
1818
pip install pytest-html
1919
echo "##[endgroup]"

0 commit comments

Comments
 (0)