Spaces:
Running
on
L40S
Running
on
L40S
File size: 643 Bytes
d7e58f0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
#!/bin/bash
# Copyright (c) Facebook, Inc. and its affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
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__)")
export BUILD_TYPE=wheel
setup_env "$VERSION"
setup_wheel_python
pip_install numpy
setup_pip_pytorch_version
download_nvidiacub_if_needed
python setup.py clean
IS_WHEEL=1 python setup.py bdist_wheel
|