Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

about knn_cuda #2

Open
lrrll opened this issue Jul 16, 2020 · 6 comments
Open

about knn_cuda #2

lrrll opened this issue Jul 16, 2020 · 6 comments

Comments

@lrrll
Copy link

lrrll commented Jul 16, 2020

hi why i run the bash test_punet.sh, it happen that "no modul named knn_cuda import KNN"? i run this code with cuda9.0 python 3.6 pytorch1.0

@lyqun
Copy link
Owner

lyqun commented Jul 16, 2020

You should install knn_cuda by running

pip install --upgrade https://github.com/unlimblue/KNN_CUDA/releases/download/0.2/KNN_CUDA-0.2-py3-none-any.whl

For more details, please refer to this repo, https://github.com/unlimblue/KNN_CUDA

Thanks for pointing out, I will update the installation guideline later.

@lrrll
Copy link
Author

lrrll commented Jul 17, 2020

yes i had resolved, but another problem increasing.
i want know whether it must need cuda 10 and torch 1.2; my environment is cuda 9 and torch 1.1; but it raise the problem like this:
PU-Net_pytorch-master$ python eval.py --gpu 0 --resume logs/punet_baseline/punet
_epoch_99.pth
Namespace(batch_size=8, gpu=0, model='punet', resume='logs/punet_baseline/punet_epoch_99.pth', up_ratio=4, use_bn=False, use_res=False, workers=4)
Traceback (most recent call last):
File "eval.py", line 26, in
from pointnet2 import pointnet2_utils as pn2_utils
File "/home/ouc/l-dataset/PU-Net_pytorch-master/pointnet2/pointnet2_utils.py", line 7, in
import pointnet2_cuda as pointnet2
ImportError: /home/ouc/.conda/envs/pytorch-1.0/lib/python3.6/site-packages/pointnet2-0.0.0-py3.6-linux-x86_64.egg/pointnet2_cuda.cpython-36m-x86_64-linux-gnu.so: undefined symbol: _ZN2at19UndefinedTensorImpl10_singletonE

baidu say it is because the version of cuda is not match torch

@lyqun
Copy link
Owner

lyqun commented Jul 17, 2020

Have you complied pointnet2 successfully, by running

cd pointnet2
python setup.py install 

If cuda and pytorch version mis-match, you may need to re-install pytorch following the official document. Refer to this page, https://pytorch.org/get-started/previous-versions/

This repo is tested with torch 1.2, but I think torch 1.1 is ok.

@xiaolongTang163
Copy link

image
image
i run this code with cuda10.0 python 3.6 pytorch1.2

@xiaolongTang163
Copy link

I have solve this problem by using Ubuntu

@Ahmoudahmed
Copy link

You should install knn_cuda by this command line
pip install --upgrade https://github.com/unlimblue/KNN_CUDA/releases/download/0.2/KNN_CUDA-0.2-py3-none-any.whl

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

4 participants