search

ma-xu / pointMLP-pytorch

[ICLR 2022 poster] Official PyTorch implementation of "Rethinking Network Design and Local Geometry in Point Cloud: A Simple Residual MLP Framework"
Apache License 2.0
501 stars 65 forks source link

Error when importing pointnet2_utils #89

Closed ardianumam closed 1 year ago

ardianumam commented 1 year ago

Hi,

Thanks for the great work! I try to run the code, and get this error: CUDA error: the provided PTX was compiled with an unsupported toolchain.

This error only happens when I import as from pointnet2_ops import pointnet2_utils (commenting it has no error). And the error also only happens when I use Nvidia Ampere arc, specifically, Nvidia A5000. Here is my other setting:

  1. Python: ver 3.9
  2. Pytorch: ver 1.9, with cuda 11.1
  3. Cuda driver in my Ubuntu machine: 11.3

Note: the pointnet2 installation goes well, without any error message

Thanks!

ma-xu commented 1 year ago

Thanks for your interest. Two suggestions: 1) please upgrade your pytorch version to 1.13+ for Ampere Arch; 2) please make sure the pytorch instsall cuda version is same as your Cuda driver version. Hope that can help.

ardianumam commented 1 year ago

Thanks much for your reply. Thanks to this PyTorch forum here, I added “8.6” in the TORCH_CUDA_ARCH_LIST here, and it solved the issue.