Error in classification_ModelNet40/models/pointmlp.py #99
Description
Hi, Thanks for your sharing!
I try to test the pointmlp model and this bug just occured:
project/pointMLP-pytorch/classification_ModelNet40/models# python pointmlp.py
===> testing pointMLP ...
Traceback (most recent call last):
File "pointmlp.py", line 366, in
out = model(data)
File "/root/miniconda3/envs/pointmlp/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "pointmlp.py", line 338, in forward
xyz, x = self.local_grouper_list[i](xyz, x.permute(0, 2, 1)) # [b,g,3] [b,g,k,d]
File "/root/miniconda3/envs/pointmlp/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "pointmlp.py", line 161, in forward
fps_idx = pointnet2_utils.furthest_point_sample(xyz, self.groups).long() # [B, npoint]
File "/root/miniconda3/envs/pointmlp/lib/python3.7/site-packages/pointnet2_ops/pointnet2_utils.py", line 54, in forward
out = _ext.furthest_point_sampling(xyz, npoint)
RuntimeError: falseINTERNAL ASSERT FAILED at "/root/autodl-tmp/project/pointMLP-pytorch/pointnet2_ops_lib/pointnet2_ops/_ext-src/src/sampling.cpp":83, please report a bug to PyTorch. CPU not supported
And then I test whether my cuda is available:
Python 3.7.16 (default, Jan 17 2023, 22:20:44)
[GCC 11.2.0] :: Anaconda, Inc. on linux
Type "help", "copyright", "credits" or "license" for more information.
import torch
torch.cuda.is_available()
True
It seems like my cuda is setting properly. Could you please help me with this confused bug? Hoping for your reply,Thansk!