search

chengkai-liu / Mamba4Rec

[RelKD'24] Mamba4Rec: Towards Efficient Sequential Recommendation with Selective State Space Models
https://arxiv.org/abs/2403.03900
MIT License
85 stars 3 forks source link

CUDA error running the code #4

Closed dom-ce closed 8 months ago

dom-ce commented 8 months ago

Hi authors, I'm encountering this error when I run the code despite having configured everything correctly.

RuntimeError: CUDA error: no kernel image is available for execution on the device
CUDA kernel errors might be asynchronously reported at some other API call, so the stacktrace below might be incorrect.
For debugging consider passing CUDA_LAUNCH_BLOCKING=1.
Compile with `TORCH_USE_CUDA_DSA` to enable device-side assertions.

I'm attaching the log, a text file containing the output of the code execution, and two others corresponding respectively to the outputs of 'nvidia-smi' and 'nvcc -V'. Thank you in advance nvcc_output.txt nvidia-smi_output.txt output.txt Mamba4Rec-ml-1m-Mar-19-2024_16-11-53-ae14e3.log

chengkai-liu commented 8 months ago

It shows that causal-conv1d cannot work correctly in your setup. You can refer to the issues in the repositories of mamba and causal-conv1d. There are two issues with the same CUDA error: https://github.com/state-spaces/mamba/issues/40 https://github.com/state-spaces/mamba/issues/85

I suggest discussing these errors within the Issues on the repository of mamba.

dom-ce commented 8 months ago

Thank you for the advice. I followed the guide showed in this issue but uninstalling causal-conv1d which is optional to use mamba-ssm.