KNN_CUDA

[markhome1]

Is there any way to solve the problem of not being able to install KNN-CUDA?

[hanryyan]

What is the problem for you? Can you show the errors?

[markhome1]

File \code/SymmCompletion/models/EdgeConv.py", line 5, in
from knn_cuda import KNN
File "/miniconda3/envs/SymmCompletion/lib/python3.11/site-packages/knn_cuda/init.py", line 38, in
_knn = load_cpp_ext("knn")
^^^^^^^^^^^^^^^^^^^
File "/miniconda3/envs/SymmCompletion/lib/python3.11/site-packages/knn_cuda/init.py", line 15, in load_cpp_ext
assert torch.cuda.is_available(), "torch.cuda.is_available() is False."
AssertionError: torch.cuda.is_available() is False.
I followed the steps and this always happened. I wrote a KNN algorithm myself and it still couldn't match

[hanryyan]

I am sorry. I write the code of os.environ['CUDA_VISIBLE_DEVICES'] = "1" in the script of main.py. Maybe your machine doesn't have GPU-1. Please delete this code in the main.py.

[markhome1]

How many GPUs did you use for training?

[hanryyan]

2~4 GPUs.

[markhome1]

How long to train?

[hanryyan]

1~2 days for RTX 4090