install torch cpu

Signed-off-by: chensuyue <[email protected]>
intel · Jul 28, 2024 · 78057a0 · 78057a0
1 parent 4454f6f
commit 78057a0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/.azure-pipelines/scripts/install_nc.sh b/.azure-pipelines/scripts/install_nc.sh
@@ -3,6 +3,10 @@
 echo -e "\n Install Neural Compressor ... "
 cd /neural-compressor
 if [[ $1 = *"3x_pt"* ]]; then
+    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