update linux wheel builds

Summary:
* Add PyTorch 1.10 + CUDA 11.1 combination.
* Change the CUDA 11.3 builds to happen in a separate docker image.
* Update connection to AWS to use the official `aws` commands instead of the wrapper which is now gone.

Reviewed By: patricklabatut

Differential Revision: D33235489

fbshipit-source-id: 56401f27c002a512ae121b3ec5911d020bfab885
This commit is contained in:
Jeremy Reizenstein
2021-12-21 05:06:33 -08:00
committed by Facebook GitHub Bot
parent b51be58f63
commit cc3259ba93
3 changed files with 24 additions and 10 deletions

View File

@@ -39,7 +39,7 @@ declare -A CONDA_CUDA_VERSIONS=(
# ["1.8.1"]="cu101 cu102 cu111"
# ["1.9.0"]="cu102 cu111"
# ["1.9.1"]="cu102 cu111"
["1.10.0"]="cu102 cu113"
["1.10.0"]="cu102 cu111 cu113"
)
@@ -58,6 +58,13 @@ do
for cu_version in ${CONDA_CUDA_VERSIONS[$pytorch_version]}
do
if [[ "cu113" == *$cu_version* ]] && [[ $SELECTED_CUDA != "$cu_version" ]]
# ^^^ CUDA versions listed here have to be built
# in their own containers.
then
continue
fi
case "$cu_version" in
cu113)
export CUDA_HOME=/usr/local/cuda-11.3/