mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
5fbdb99aec
Summary: Add builds corresponding to the new pytorch 1.10.0. We omit CUDA 11.3 testing because it fails with current hardware, and omit the main build too for the moment. Also move to the newer GPU circle CI executors. Reviewed By: gkioxari Differential Revision: D32335934 fbshipit-source-id: 416d92a8eecd06ef7fc742664a5f2d46f93415f8
124 lines
5.3 KiB
Bash
124 lines
5.3 KiB
Bash
#!/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
|
|
|
|
conda init bash
|
|
# shellcheck source=/dev/null
|
|
source ~/.bashrc
|
|
|
|
cd /inside
|
|
VERSION=$(python -c "exec(open('pytorch3d/__init__.py').read()); print(__version__)")
|
|
|
|
export BUILD_VERSION=$VERSION
|
|
export FORCE_CUDA=1
|
|
|
|
wget --no-verbose https://github.com/NVIDIA/cub/archive/1.10.0.tar.gz
|
|
tar xzf 1.10.0.tar.gz
|
|
CUB_HOME=$(realpath ./cub-1.10.0)
|
|
export CUB_HOME
|
|
echo "CUB_HOME is now $CUB_HOME"
|
|
|
|
# As a rule, we want to build for any combination of dependencies which is supported by
|
|
# PyTorch3D and not older than the current Google Colab set up.
|
|
|
|
PYTHON_VERSIONS="3.7 3.8 3.9"
|
|
# the keys are pytorch versions
|
|
declare -A CONDA_CUDA_VERSIONS=(
|
|
# ["1.4.0"]="cu101"
|
|
# ["1.5.0"]="cu101 cu102"
|
|
# ["1.5.1"]="cu101 cu102"
|
|
# ["1.6.0"]="cu101 cu102"
|
|
# ["1.7.0"]="cu101 cu102 cu110"
|
|
# ["1.7.1"]="cu101 cu102 cu110"
|
|
# ["1.8.0"]="cu101 cu102 cu111"
|
|
# ["1.8.1"]="cu101 cu102 cu111"
|
|
["1.9.0"]="cu102 cu111"
|
|
["1.9.1"]="cu102 cu111"
|
|
)
|
|
|
|
|
|
|
|
for python_version in $PYTHON_VERSIONS
|
|
do
|
|
for pytorch_version in "${!CONDA_CUDA_VERSIONS[@]}"
|
|
do
|
|
if [[ "3.6 3.7 3.8" != *$python_version* ]] && [[ "1.4.0 1.5.0 1.5.1 1.6.0 1.7.0" == *$pytorch_version* ]]
|
|
then
|
|
#python 3.9 and later not supported by pytorch 1.7.0 and before
|
|
continue
|
|
fi
|
|
|
|
extra_channel="-c conda-forge"
|
|
|
|
for cu_version in ${CONDA_CUDA_VERSIONS[$pytorch_version]}
|
|
do
|
|
case "$cu_version" in
|
|
cu113)
|
|
export CUDA_HOME=/usr/local/cuda-11.3/
|
|
export CUDA_TAG=11.3
|
|
export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_86,code=sm_86 -gencode=arch=compute_50,code=compute_50"
|
|
;;
|
|
cu112)
|
|
export CUDA_HOME=/usr/local/cuda-11.2/
|
|
export CUDA_TAG=11.2
|
|
export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_86,code=sm_86 -gencode=arch=compute_50,code=compute_50"
|
|
;;
|
|
cu111)
|
|
export CUDA_HOME=/usr/local/cuda-11.1/
|
|
export CUDA_TAG=11.1
|
|
export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_86,code=sm_86 -gencode=arch=compute_50,code=compute_50"
|
|
;;
|
|
cu110)
|
|
export CUDA_HOME=/usr/local/cuda-11.0/
|
|
export CUDA_TAG=11.0
|
|
export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_80,code=sm_80 -gencode=arch=compute_50,code=compute_50"
|
|
;;
|
|
cu102)
|
|
export CUDA_HOME=/usr/local/cuda-10.2/
|
|
export CUDA_TAG=10.2
|
|
export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_50,code=compute_50"
|
|
;;
|
|
cu101)
|
|
export CUDA_HOME=/usr/local/cuda-10.1/
|
|
export CUDA_TAG=10.1
|
|
export NVCC_FLAGS="-gencode=arch=compute_35,code=sm_35 -gencode=arch=compute_50,code=sm_50 -gencode=arch=compute_60,code=sm_60 -gencode=arch=compute_70,code=sm_70 -gencode=arch=compute_75,code=sm_75 -gencode=arch=compute_50,code=compute_50"
|
|
;;
|
|
*)
|
|
echo "Unrecognized cu_version=$cu_version"
|
|
exit 1
|
|
;;
|
|
esac
|
|
tag=py"${python_version//./}"_"${cu_version}"_pyt"${pytorch_version//./}"
|
|
|
|
outdir="/inside/packaging/linux_wheels/output/$tag"
|
|
if [[ -d "$outdir" ]]
|
|
then
|
|
continue
|
|
fi
|
|
|
|
conda create -y -n "$tag" "python=$python_version"
|
|
conda activate "$tag"
|
|
conda install -y -c pytorch $extra_channel "pytorch=$pytorch_version" "cudatoolkit=$CUDA_TAG" torchvision
|
|
pip install fvcore iopath
|
|
echo "python version" "$python_version" "pytorch version" "$pytorch_version" "cuda version" "$cu_version" "tag" "$tag"
|
|
|
|
rm -rf dist
|
|
|
|
python setup.py clean
|
|
python setup.py bdist_wheel
|
|
|
|
rm -rf "$outdir"
|
|
mkdir -p "$outdir"
|
|
cp dist/*whl "$outdir"
|
|
|
|
conda deactivate
|
|
done
|
|
done
|
|
done
|
|
echo "DONE"
|