Skip to content
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.

Commit 95bac7c

Browse files
committedFeb 11, 2025·
update torch installation
Signed-off-by: Sun, Xuehao <xuehao.sun@intel.com>
1 parent 5a33c16 commit 95bac7c

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed
 

‎.azure-pipelines/scripts/install_nc.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ if [[ $1 = *"3x_pt"* ]]; then
99
python setup.py pt bdist_wheel
1010
else
1111
echo -e "\n Install torch CPU ... "
12-
pip install torch==2.5.1
12+
pip install torch==2.5.1 --index-url https://download.pytorch.org/whl/cpu
1313
python -m pip install intel-extension-for-pytorch==2.5.0 oneccl_bind_pt==2.5.0 --extra-index-url https://pytorch-extension.intel.com/release-whl/stable/cpu/us/
1414
python -m pip install --no-cache-dir -r requirements.txt
1515
python setup.py bdist_wheel

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

+1
Original file line numberDiff line numberDiff line change
@@ -13,6 +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
1617
pip install pytest-cov
1718
pip install pytest-html
1819
echo "##[endgroup]"

0 commit comments

Comments
 (0)
Please sign in to comment.