No module named 'frnn'

[narges-tk]

Hi! Thanks for publishing your code! While everything went well in installation phase, now I am getting No module named 'frnn' error. Any help would be appreciated. Please find the complete error below.

ntakhtke  ❲c❳ spt  ~/DL_models/superpoint_transformer-master  1  python src/train.py experiment=semantic/narges_data
Traceback (most recent call last):
File "src/train.py", line 49, in
from src import utils
File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/init.py", line 2, in
import src.data
File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/data/init.py", line 1, in
from .csr import *
File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/data/csr.py", line 5, in
from src.utils import tensor_idx, is_sorted, indices_to_pointers,
File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/utils/init.py", line 12, in
from .neighbors import *
File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/utils/neighbors.py", line 3, in
from src.dependencies.FRNN import frnn
File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/dependencies/FRNN/frnn/init.py", line 2, in
from .frnn import frnn_grid_points, frnn_gather, frnn_bf_points, _C
File "/home/ntakhtke/DL_models/superpoint_transformer-master/src/dependencies/FRNN/frnn/frnn.py", line 7, in
from frnn import _C
ModuleNotFoundError: No module named 'frnn'

[drprojects]

Hi @narges-tk, thanks for your interest in this project.

This error comes from the fact that the dependency FRNN was not properly installed when running install.sh.

This may be related to #77, please check this first.

PS: if you ❤️ or use this project, don't forget to give it a ⭐, it means a lot to us !

[narges-tk]

Thanks for your prompt reply.
I resolved it by setting CUDA_HOME as follows:
export CUDA_HOME=/usr/local/cuda-12.1