From 74e9381213e931524f25080773d7644c7f739bc9 Mon Sep 17 00:00:00 2001 From: chensuyue Date: Sun, 28 Jul 2024 15:40:51 +0800 Subject: [PATCH] install torch cpu Signed-off-by: chensuyue --- .azure-pipelines/scripts/install_nc.sh | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/.azure-pipelines/scripts/install_nc.sh b/.azure-pipelines/scripts/install_nc.sh index 0caf4b1a651..6a80419259f 100644 --- a/.azure-pipelines/scripts/install_nc.sh +++ b/.azure-pipelines/scripts/install_nc.sh @@ -3,12 +3,13 @@ echo -e "\n Install Neural Compressor ... " cd /neural-compressor if [[ $1 = *"3x_pt"* ]]; then - python -m pip install --no-cache-dir -r requirements_pt.txt - python setup.py pt bdist_wheel - pip install --no-deps dist/neural_compressor*.whl --force-reinstall if [[ $1 != *"3x_pt_fp8"* ]]; then + echo -e "\n Install torch CPU ... " pip install torch==2.3.0 --index-url https://download.pytorch.org/whl/cpu fi + python -m pip install --no-cache-dir -r requirements_pt.txt + python setup.py pt bdist_wheel + pip install --no-deps dist/neural_compressor*.whl --force-reinstall elif [[ $1 = *"3x_tf"* ]]; then python -m pip install --no-cache-dir -r requirements_tf.txt python setup.py tf bdist_wheel