NVIDIA / MinkowskiEngine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors
https://nvidia.github.io/MinkowskiEngine
Other
2.43k stars 360 forks source link

Error when calling MinkowskiPooling #565

Closed LinfengUP closed 11 months ago

LinfengUP commented 11 months ago

Describe the bug When I apply MinkowskiPooling, it returns \<unknown> error without any extra information. 6dc7b4d49ff4dc34a7bc0822f002d5e


To Reproduce

model.pooling = MinkowskiAvgPooling(
            kernel_size=2, stride=2, dimension=3)

with torch.no_grad():
    coordinates = ME.SparseTensor(
        features=raw_coordinates,
        coordinate_manager=aux[-1].coordinate_manager,
        coordinate_map_key=aux[-1].coordinate_map_key,
        device=aux[-1].device)
    coords = [coordinates]
    for _ in reversed(range(len(aux) - 1)):
        coords.append(model.pooling(coords[-1]))
    coords.reverse()

Expected behavior A clear and concise description of what you expected to happen.


Desktop (please complete the following information):

Jovendish commented 10 months ago

I encountered the same problem. Has the problem been solved?

LinfengUP commented 10 months ago

I encountered the same problem. Has the problem been solved?

Oh, I realized I did not add CUDA_HOME in environment variables. Just check whether your nvcc -V can work well.