build_conda.sh 966 Bytes
Newer Older
facebook-github-bot's avatar
facebook-github-bot committed
1
2
3
4
5
6
7
#!/bin/bash
# Copyright (c) Facebook, Inc. and its affiliates. All rights reserved.
set -ex

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

8
9
VERSION=$(python -c "exec(open('${script_dir}/../pytorch3d/__init__.py').read()); print(__version__)")

10
11
12
# Prevent dev tag in the version string.
export BUILD_VERSION=$VERSION

facebook-github-bot's avatar
facebook-github-bot committed
13
export BUILD_TYPE=conda
14
setup_env "$VERSION"
facebook-github-bot's avatar
facebook-github-bot committed
15
16
17
18
export SOURCE_ROOT_DIR="$PWD"
setup_conda_pytorch_constraint
setup_conda_cudatoolkit_constraint
setup_visual_studio_constraint
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
19
20
21
22
23
24
25
26
27

if [[ "$JUST_TESTRUN" == "1" ]]
then
    # We are not building for other users, we
    # are only trying to see if the tests pass.
    # So save time by only building for our own GPU.
    unset NVCC_FLAGS
fi

facebook-github-bot's avatar
facebook-github-bot committed
28
# shellcheck disable=SC2086
Jeremy Reizenstein's avatar
Jeremy Reizenstein committed
29
conda build $CONDA_CHANNEL_FLAGS ${TEST_FLAG:-} -c bottler -c defaults -c fvcore -c iopath -c conda-forge --no-anaconda-upload --python "$PYTHON_VERSION" packaging/pytorch3d