mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-02 03:42:50 +08:00
c639198c97
Summary: Build for pytorch 1.9, and make it the only mac build. Not testing on cuda 11.1, because of annoying failures which are restricted to certain hardware. Also update cuda driver in CI tests. Reviewed By: patricklabatut Differential Revision: D29302314 fbshipit-source-id: 78def378adb9d7aa287abdc5ac0af269e3ba3625
118 lines
4.4 KiB
Bash
118 lines
4.4 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"
|
|
)
|
|
|
|
|
|
|
|
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
|
|
|
|
if [[ "3.9" == "$python_version" ]]
|
|
then
|
|
extra_channel="-c conda-forge"
|
|
else
|
|
extra_channel=""
|
|
fi
|
|
|
|
for cu_version in ${CONDA_CUDA_VERSIONS[$pytorch_version]}
|
|
do
|
|
case "$cu_version" in
|
|
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"
|