Skip to content

Commit 78057a0

Browse files
committed
install torch cpu
Signed-off-by: chensuyue <[email protected]>
1 parent 4454f6f commit 78057a0

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

.azure-pipelines/scripts/install_nc.sh

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,10 @@
33
echo -e "\n Install Neural Compressor ... "
44
cd /neural-compressor
55
if [[ $1 = *"3x_pt"* ]]; then
6+
if [[ $1 != *"3x_pt_fp8"* ]]; then
7+
echo -e "\n Install torch CPU ... "
8+
pip install torch==2.3.0 --index-url https://download.pytorch.org/whl/cpu
9+
fi
610
python -m pip install --no-cache-dir -r requirements_pt.txt
711
python setup.py pt bdist_wheel
812
pip install --no-deps dist/neural_compressor*.whl --force-reinstall

0 commit comments

Comments
 (0)