search

zyc00 / Point-SAM

Point-SAM: This is the official repository of "Point-SAM: Promptable 3D Segmentation Model for Point Clouds". We provide codes for running our demo and links to download checkpoints.
MIT License
143 stars 9 forks source link

module 'torkit3d._C' has no attribute 'sample_farthest_points_cuda' #2 #10

Closed pyr0ken closed 1 week ago

pyr0ken commented 2 months ago

Hello, I get this error when using sample_farthest_points:

.../.local/lib/python3.10/site-packages/torkit3d/ops/sample_farthest_points.py", line 13, in forward
    index = _C.sample_farthest_points_cuda(points, num_samples)
AttributeError: module 'torkit3d._C' has no attribute 'sample_farthest_points_cuda'

I installed torkit3d using this command : 

pip install "git+https://github.com/Jiayuan-Gu/torkit3d.git"

This is the installed package‌ : image

Jiayuan-Gu commented 2 months ago

Please add FORCE_CUDA=1 as the env variable: FORCE_CUDA=1 pip install "git+https://github.com/Jiayuan-Gu/torkit3d.git". Or you can use replace torkit3d with pytorch3d for the FPS (farthest point sampling).

pyr0ken commented 2 months ago

Thank you !