423A35C7/pytorch3d
1
0
Fork 0
mirror of https://github.com/facebookresearch/pytorch3d.git synced 2025-12-14 11:26:24 +08:00
Code Issues Packages Projects Releases Wiki Activity

builds for PyTorch 1.9

Browse Source 
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
This commit is contained in:
Jeremy Reizenstein
2021-06-22 12:38:36 -07:00
committed by Facebook GitHub Bot
parent bbc12e70c4
commit c639198c97
4 changed files with 71 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
packaging/linux_wheels/inside.sh
View File

@@ -23,8 +23,10 @@ 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.6 3.7 3.8 3.9"
PYTHON_VERSIONS="3.7 3.8 3.9"
# the keys are pytorch versions
declare -A CONDA_CUDA_VERSIONS=(
# ["1.4.0"]="cu101"
@@ -33,8 +35,9 @@ declare -A CONDA_CUDA_VERSIONS=(
# ["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.8.0"]="cu101 cu102 cu111"
# ["1.8.1"]="cu101 cu102 cu111"
["1.9.0"]="cu102 cu111"
)