diff --git a/pointnet2_ops_lib/setup.py b/pointnet2_ops_lib/setup.py index faf7154..75807be 100644 --- a/pointnet2_ops_lib/setup.py +++ b/pointnet2_ops_lib/setup.py @@ -16,7 +16,7 @@ exec(open(osp.join("pointnet2_ops", "_version.py")).read()) -os.environ["TORCH_CUDA_ARCH_LIST"] = "3.7+PTX;5.0;6.0;6.1;6.2;7.0;7.5" +os.environ["TORCH_CUDA_ARCH_LIST"] = "5.0;6.0;6.1;6.2;7.0;7.5;8.0;8.6;8.7;8.9;9.0+PTX" setup( name="pointnet2_ops", version=__version__,