Skip to content

Commit 1afef1d

Browse files
authored
OS X wheel updates for nightlies (#1205)
Signed-off-by: Edward Z. Yang <[email protected]>
1 parent 6c7189f commit 1afef1d

File tree

1 file changed

+11
-3
lines changed

1 file changed

+11
-3
lines changed

packaging/wheel/osx_wheel.sh

Lines changed: 11 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -12,15 +12,15 @@ wget -q https://repo.anaconda.com/miniconda/Miniconda3-latest-MacOSX-x86_64.sh &
1212
. ~/minconda_wheel_env_tmp/bin/activate
1313

1414

15-
export TORCHVISION_BUILD_VERSION="0.3.0"
15+
export TORCHVISION_BUILD_VERSION="0.4.0.dev$(date "+%Y%m%d")"
1616
export TORCHVISION_BUILD_NUMBER="1"
1717
export OUT_DIR=~/torchvision_wheels
1818

1919
export MACOSX_DEPLOYMENT_TARGET=10.9 CC=clang CXX=clang++
2020

2121
pushd /tmp
2222
rm -rf vision
23-
git clone https://github.com/pytorch/vision -b v${TORCHVISION_BUILD_VERSION}
23+
git clone https://github.com/pytorch/vision
2424
pushd vision
2525

2626
desired_pythons=( "2.7" "3.5" "3.6" "3.7" )
@@ -32,8 +32,16 @@ do
3232
conda create -yn $env_name python="$desired_python"
3333
conda activate $env_name
3434

35+
pip uninstall -y torch || true
36+
pip uninstall -y torch_nightly || true
37+
38+
export TORCHVISION_PYTORCH_DEPENDENCY_NAME=torch_nightly
39+
pip install torch_nightly -f https://download.pytorch.org/whl/nightly/cpu/torch_nightly.html
40+
export TORCHVISION_PYTORCH_DEPENDENCY_VERSION="$(pip show torch_nightly | grep ^Version: | sed 's/Version: *//')"
41+
echo "Building against ${TORCHAUDIO_PYTORCH_DEPENDENCY_VERSION}"
42+
3543
# install torchvision dependencies
36-
pip install torch ninja scipy pytest
44+
pip install ninja scipy pytest
3745

3846
python setup.py clean
3947
python setup.py bdist_wheel

0 commit comments

Comments
 (0)