File tree 2 files changed +2
-1
lines changed
2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -9,7 +9,7 @@ if [[ $1 = *"3x_pt"* ]]; then
9
9
python setup.py pt bdist_wheel
10
10
else
11
11
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
13
13
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/
14
14
python -m pip install --no-cache-dir -r requirements.txt
15
15
python setup.py bdist_wheel
Original file line number Diff line number Diff line change @@ -13,6 +13,7 @@ echo "##[section]import check pass"
13
13
echo " ##[group]set up UT env..."
14
14
export LD_LIBRARY_PATH=/usr/local/lib/:$LD_LIBRARY_PATH
15
15
pip install -r /neural-compressor/test/3x/torch/requirements.txt
16
+ pip install torch==2.5.1 # For auto-round
16
17
pip install pytest-cov
17
18
pip install pytest-html
18
19
echo " ##[endgroup]"
You can’t perform that action at this time.
0 commit comments