diff --git a/.azure-pipelines/scripts/install_nc.sh b/.azure-pipelines/scripts/install_nc.sh index 55b323c56c2..dc211d0f512 100644 --- a/.azure-pipelines/scripts/install_nc.sh +++ b/.azure-pipelines/scripts/install_nc.sh @@ -6,6 +6,9 @@ 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 + pip install torch --index-url https://download.pytorch.org/whl/cpu + fi elif [[ $1 = *"3x_tf"* ]]; then python -m pip install --no-cache-dir -r requirements_tf.txt python setup.py tf bdist_wheel diff --git a/README.md b/README.md index a10e98d61e7..1568706994e 100644 --- a/README.md +++ b/README.md @@ -36,7 +36,7 @@ support AMD CPU, ARM CPU, and NVidia GPU through ONNX Runtime with limited testi ```Shell pip install torch --index-url https://download.pytorch.org/whl/cpu ``` -#### Install torch for XPU +#### Install torch/intel_extension_for_pytorch for XPU https://intel.github.io/intel-extension-for-pytorch/index.html#installation #### Install torch for other platform