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

pytorch 2.1, drop mac builds

Browse Source 
Summary: Build updates for PyTorch 2.1

Reviewed By: MichaelRamamonjisoa

Differential Revision: D50345762

fbshipit-source-id: 89bf4edf1c21566aa86a3abca9b4df7c4d1d17a2
This commit is contained in:
Jeremy Reizenstein
2023-10-17 06:04:06 -07:00
committed by Facebook GitHub Bot
parent eaf0709d6a
commit 28f914bf3b
6 changed files with 67 additions and 266 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
packaging/build_conda.py
View File

@@ -59,12 +59,17 @@ def setup_cuda():
)
if CU_VERSION == "cu102":
nvcc_flags = basic_nvcc_flags
elif CU_VERSION == "cu110":
nvcc_flags = "-gencode=arch=compute_80,code=sm_80 " + basic_nvcc_flags
elif CU_VERSION < ("cu118"):
nvcc_flags = (
"-gencode=arch=compute_80,code=sm_80 "
+ "-gencode=arch=compute_86,code=sm_86 "
+ basic_nvcc_flags
)
else:
nvcc_flags = (
"-gencode=arch=compute_80,code=sm_80 "
+ "-gencode=arch=compute_86,code=sm_86 "
+ "-gencode=arch=compute_90,code=sm_90 "
+ basic_nvcc_flags
)