-
Notifications
You must be signed in to change notification settings - Fork 445
Open
Description
My configuration:
ubuntu20.04
4060
CUDA11.1(NVCC -V)
pytorch1.9.0(or1.10.0) cu11.1
The following error was encountered when running the demo.py:
Traceback (most recent call last):
File "/opt/project/regtr/src/demo.py", line 196, in <module>
main()
File "/opt/project/regtr/src/demo.py", line 182, in main
outputs = model(data_batch)
File "/opt/conda/envs/regtr_pytorch1.10.0/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/opt/project/regtr/src/models/regtr.py", line 117, in forward
kpconv_meta = self.preprocessor(batch['src_xyz'] + batch['tgt_xyz'])
File "/opt/conda/envs/regtr_pytorch1.10.0/lib/python3.8/site-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/opt/project/regtr/src/models/backbone_kpconv/kpconv.py", line 489, in forward
pool_p, pool_b = batch_grid_subsampling_kpconv_gpu(
File "/opt/project/regtr/src/models/backbone_kpconv/kpconv.py", line 232, in batch_grid_subsampling_kpconv_gpu
sparse_tensor = ME.SparseTensor(
File "/opt/conda/envs/regtr_pytorch1.10.0/lib/python3.8/site-packages/MinkowskiEngine-0.5.4-py3.8-linux-x86_64.egg/MinkowskiEngine/MinkowskiSparseTensor.py", line 271, in __init__
coordinates, features, coordinate_map_key = self.initialize_coordinates(
File "/opt/conda/envs/regtr_pytorch1.10.0/lib/python3.8/site-packages/MinkowskiEngine-0.5.4-py3.8-linux-x86_64.egg/MinkowskiEngine/MinkowskiSparseTensor.py", line 334, in initialize_coordinates
features = spmm_avg.apply(self.inverse_mapping, cols, size, features)
File "/opt/conda/envs/regtr_pytorch1.10.0/lib/python3.8/site-packages/MinkowskiEngine-0.5.4-py3.8-linux-x86_64.egg/MinkowskiEngine/sparse_matrix_functions.py", line 183, in forward
result, COO, vals = spmm_average(
File "/opt/conda/envs/regtr_pytorch1.10.0/lib/python3.8/site-packages/MinkowskiEngine-0.5.4-py3.8-linux-x86_64.egg/MinkowskiEngine/sparse_matrix_functions.py", line 93, in spmm_average
result, COO, vals = MEB.coo_spmm_average_int32(
RuntimeError: <unknown> at /opt/project/MinkowskiEngine-0.5.4.1/src/gpu.cu:100
Can anyone help me solve it?
Metadata
Metadata
Assignees
Labels
No labels