search

isyangshu / MambaMIL

[MICCAI 2024] Official Code for "MambaMIL: Enhancing Long Sequence Modeling with Sequence Reordering in Computational Pathology"
49 stars 1 forks source link

Regarding the package import #15

Closed ZhengtingJiang closed 4 weeks ago

ZhengtingJiang commented 1 month ago

Hi! I am encountering a problem and I cannot solve it. 屏幕截图 2024-06-12 105719

Could you pls give some advice?

wyhsleep commented 1 month ago

What is your CUDA version?

ZhengtingJiang commented 1 month ago

Hi! It is Ubuntu Server 22.04 LTS R535 CUDA 12.2

wyhsleep commented 1 month ago

Maybe you can try CUDA11.8. The code works fine in that version and we will release a version that supports CUDA12.X in the near future. If you have further problems, feel free to contact us.

ZhengtingJiang commented 1 month ago

Thank you! I would have a try.