NVIDIA / MinkowskiEngine

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

RuntimeError: invalid argument at /tmp/pip-req-build-24zubhoq/src/spmm.cu:590 #426

Open JimmyWong1998 opened 2 years ago

JimmyWong1998 commented 2 years ago

Hi, I met the following problem:

2bfb933cc72bfbbfbd1dc0537642647

My situation is different from #421. His CUDA version is 11.1 but my CUDA version is 10.2. I run:

_python -m examples.classificationmodelnet40

with both GCC 7.4 and GCC 7.5. They both return that RuntimeError: invalid argument at /tmp/pip-req-build-24zubhoq/src/spmm.cu:590

Could you help me how to fix this error? Thank you very much!

Desktop (please complete the following information):

OS: Linux GCC: 7.5.0 Python version: 3.8.12 Pytorch version: 1.91 CUDA version: 10.2 NVIDIA Driver version: 440.33.01 Minkowski Engine version 0.5.4

weiguangzhao commented 2 years ago

Try GCC 9.3.0

JimmyWong1998 commented 2 years ago

Thank you for your suggestion @weiguangzhao. However, in my environment, the install guidance says GCC must lower than 8.

And I run the code in Server which haven't the driver for CUDA 11.x. Have you tried to run the classification_modelnet40 with MinkowskiEngine successfully on CUDA 10.2?