diff --git a/.github/actions/setup-pytorch/action.yml b/.github/actions/setup-pytorch/action.yml index b6cc894722..88f482f78a 100644 --- a/.github/actions/setup-pytorch/action.yml +++ b/.github/actions/setup-pytorch/action.yml @@ -83,7 +83,7 @@ runs: uses: ./.github/actions/load env: # Increase this value to reset cache - CACHE_NUMBER: 12 + CACHE_NUMBER: 14 with: path: pytorch key: pytorch-$PYTORCH_CACHE_KEY-$CACHE_NUMBER @@ -120,7 +120,7 @@ runs: cd pytorch pip install wheel pip install -r requirements.txt - python setup.py bdist_wheel + USE_STATIC_MKL=1 python setup.py bdist_wheel - name: Install PyTorch (built from source) if: ${{ inputs.mode == 'source' }} diff --git a/scripts/compile-pytorch-ipex.sh b/scripts/compile-pytorch-ipex.sh index 7c5a41f6c8..d753abe113 100755 --- a/scripts/compile-pytorch-ipex.sh +++ b/scripts/compile-pytorch-ipex.sh @@ -117,7 +117,7 @@ if [[ $BUILD_PYTORCH = true ]]; then echo "****** Building $PYTORCH_PROJ ******" pip install -r requirements.txt pip install cmake ninja "numpy<2.0" - python setup.py bdist_wheel + USE_STATIC_MKL=1 python setup.py bdist_wheel echo "****** Installing PyTorch ******" pip install dist/*.whl diff --git a/scripts/install-pytorch.sh b/scripts/install-pytorch.sh index a8a0f2b83a..74b3ac5158 100755 --- a/scripts/install-pytorch.sh +++ b/scripts/install-pytorch.sh @@ -155,7 +155,7 @@ $SCRIPTS_DIR/patch-pytorch.sh echo "****** Building $PYTORCH_PROJ ******" pip install -r requirements.txt pip install cmake ninja -python setup.py bdist_wheel +USE_STATIC_MKL=1 python setup.py bdist_wheel echo "****** Installing PyTorch ******" pip install dist/*.whl