build_wheel.sh 550 Bytes
Newer Older
1
2
3
#!/bin/bash
set -ex

4
5
echo FFMPEG_ROOT=${FFMPEG_ROOT}

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

9
export BUILD_TYPE="wheel"
10
setup_env
Edward Z. Yang's avatar
Edward Z. Yang committed
11
setup_wheel_python
moto's avatar
moto committed
12
pip_install numpy future cmake ninja
13
14
setup_pip_pytorch_version
python setup.py clean
peterjc123's avatar
peterjc123 committed
15
16
if [[ "$OSTYPE" == "msys" ]]; then
    python_tag="$(echo "cp$PYTHON_VERSION" | tr -d '.')"
17
    "$script_dir/vc_env_helper.bat" python setup.py bdist_wheel --plat-name win_amd64 --python-tag $python_tag
peterjc123's avatar
peterjc123 committed
18
else
19
    python setup.py bdist_wheel
peterjc123's avatar
peterjc123 committed
20
fi