frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update build_wheel.sh

Browse Source
This commit is contained in:
Nikita Shulga
2025-01-28 15:14:41 -08:00
committed by GitHub
parent eea7d395e5
commit 049f042e52
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
.ci/wheel/build_wheel.sh
View File

@ -190,9 +190,8 @@ retry pip install -r "${pytorch_rootdir}/requirements.txt" || true
retry pip uninstall -y cmake
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -y llvm-openmp=14.0.6 cmake
# For USE_DISTRIBUTED=1 on macOS, need libuv and pkg-config to find libuv.
# For USE_DISTRIBUTED=1 on macOS, need libuv, which is build as part of tensorpipe submodule
export USE_DISTRIBUTED=1
retry conda install ${EXTRA_CONDA_INSTALL_FLAGS} -y libuv pkg-config
if [[ -n "$CROSS_COMPILE_ARM64" ]]; then
export CMAKE_OSX_ARCHITECTURES=arm64