diff --git a/README.rst b/README.rst index d53f6bf..c654eb7 100644 --- a/README.rst +++ b/README.rst @@ -1,6 +1,7 @@ Pointnet2/Pointnet++ PyTorch ============================ - +*** Compilation problem solved with pytorch2.2.1 with CUDA12.1 on Ubuntu22.04 +*** If still have issue about ninja, install build-essential and ninja-build or change “cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)}” in pointnet2_ops_lib/setup.py **Project Status**: Unmaintained. Due to finite time, I have no plans to update this code and I will not be responding to issues. diff --git a/pointnet2_ops_lib/setup.py b/pointnet2_ops_lib/setup.py index faf7154..e6bbbd9 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" setup( name="pointnet2_ops", version=__version__,