install torch cpu

Signed-off-by: chensuyue <[email protected]>
intel · Jul 28, 2024 · 74e9381 · 74e9381
1 parent 4454f6f
commit 74e9381
Showing 1 changed file with 4 additions and 3 deletions.
diff --git 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