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

PyTorch 1.8 builds

Browse Source 
Summary: Nightly builds to support PyTorch 1.8.0 and PyTorch 1.8.1.

Reviewed By: patricklabatut

Differential Revision: D29098081

fbshipit-source-id: fc6b36e919892ea41979a03e64a6fc8003528b78
This commit is contained in:
Jeremy Reizenstein
2021-06-14 10:26:44 -07:00
committed by Facebook GitHub Bot
parent 780e231536
commit 9de627e01b
7 changed files with 186 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
packaging/linux_wheels/inside.sh
View File

@@ -26,8 +26,10 @@ declare -A CONDA_CUDA_VERSIONS=(
# ["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.7.0"]="cu101 cu102 cu110"
# ["1.7.1"]="cu101 cu102 cu110"
["1.8.0"]="cu101 cu102 cu111"
["1.8.1"]="cu101 cu102 cu111"
)
@@ -52,6 +54,11 @@ do
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