mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-12-20 14:20:38 +08:00
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:
committed by
Facebook GitHub Bot
parent
b51be58f63
commit
cc3259ba93
@@ -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/
|
||||
|
||||
Reference in New Issue
Block a user