build_wheel.sh 641 Bytes
Newer Older
1
2
3
4
5
6
#!/bin/bash
set -ex

script_dir="$( cd "$( dirname "${BASH_SOURCE[0]}" )" >/dev/null 2>&1 && pwd )"
. "$script_dir/pkg_helpers.bash"

7
export BUILD_TYPE="wheel"
Edward Z. Yang's avatar
Edward Z. Yang committed
8
export NO_CUDA_PACKAGE=1
moto's avatar
moto committed
9
setup_env 0.6.0
Edward Z. Yang's avatar
Edward Z. Yang committed
10
setup_wheel_python
peterjc123's avatar
peterjc123 committed
11
12
13
if [[ "$OSTYPE" != "msys" ]]; then
    "$script_dir/build_from_source.sh" "$(pwd)"  # Build static dependencies
fi
14
15
16
pip_install numpy future
setup_pip_pytorch_version
python setup.py clean
peterjc123's avatar
peterjc123 committed
17
18
19
20
21
22
if [[ "$OSTYPE" == "msys" ]]; then
    python_tag="$(echo "cp$PYTHON_VERSION" | tr -d '.')"
    IS_WHEEL=1 python setup.py bdist_wheel --plat-name win_amd64 --python-tag $python_tag
else
    IS_WHEEL=1 python setup.py bdist_wheel
fi