mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
Summary: Update `main` build to latest CircleCI image - Ubuntu 2020.04. Avoid torch.logical_or and logical_and for PyTorch 1.4 compatibility. Also speed up the test run with Pytorch 1.4.0 (which has no ninja) by not setting NVCC_FLAGS for it. Reviewed By: theschnitz Differential Revision: D27262327 fbshipit-source-id: ddc359d134b1dc755f8b20bd3f33bb080cb3a0e1
30 lines
966 B
Bash
Executable File
30 lines
966 B
Bash
Executable File
#!/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"
|
|
|
|
VERSION=$(python -c "exec(open('${script_dir}/../pytorch3d/__init__.py').read()); print(__version__)")
|
|
|
|
# Prevent dev tag in the version string.
|
|
export BUILD_VERSION=$VERSION
|
|
|
|
export BUILD_TYPE=conda
|
|
setup_env "$VERSION"
|
|
export SOURCE_ROOT_DIR="$PWD"
|
|
setup_conda_pytorch_constraint
|
|
setup_conda_cudatoolkit_constraint
|
|
setup_visual_studio_constraint
|
|
|
|
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
|
|
|
|
# shellcheck disable=SC2086
|
|
conda build $CONDA_CHANNEL_FLAGS ${TEST_FLAG:-} -c bottler -c defaults -c conda-forge -c fvcore -c iopath --no-anaconda-upload --python "$PYTHON_VERSION" packaging/pytorch3d
|