Skip to content

Commit d8f73fe

Browse files
authored
Build PyTorch wheels (#4454)
This reverts accidental change from 19c32a7. We need to use PyTorch wheels to cache them, for example, on Windows runners. Signed-off-by: Pavel Chekin <[email protected]>
1 parent 184f6db commit d8f73fe

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

scripts/install-pytorch.sh

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -199,13 +199,13 @@ function build_pytorch {
199199
pip install 'cmake<4.0.0'
200200
pip install -r requirements.txt
201201
pip install cmake ninja
202-
USE_XCCL=1 USE_STATIC_MKL=1 python setup.py develop
202+
USE_XCCL=1 USE_STATIC_MKL=1 python setup.py bdist_wheel
203203
}
204204

205205
function install_pytorch {
206206
echo "****** Installing PyTorch ******"
207207
cd "$PYTORCH_PROJ"
208-
# pip install dist/*.whl
208+
pip install dist/*.whl
209209
}
210210

211211
if [ "$CHECK_WHEEL" = false ] || ! pytorch_wheel_exists; then

0 commit comments

Comments
 (0)