erikwijmans / Pointnet2_PyTorch

PyTorch implementation of Pointnet2/Pointnet++
The Unlicense
1.53k stars 347 forks source link

Error when importing pointnet2_utils #175

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 related to this code 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 installation goes well, without any error message

Thanks!

ardianumam commented 1 year ago

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