Can I run it with low version cuda ?

MzeroMiko / VMamba

VMamba: Visual State Space Models，code is based on mamba

MIT License

1.83k stars 100 forks source link

Can I run it with low version cuda ? #176

Open awei0116 opened 2 months ago

awei0116 commented 2 months ago

Hello author , when I Run the command “cd kernels/selective_scan && pip install .”，it occured that “RuntimeError: package is only supported on CUDA 11.6 and above. Note: make sure nvcc has a supported version by running nvcc -V”. So can I run it with cuda=11.3?What should I revise the code？My pytorch version is "torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1"
looking forward to your responding! fab6a26a98cece1389f047f8f78bdd7

MzeroMiko commented 2 months ago

I have tested the code in cuda11.3 and it works well expect for the possible speed difference or performance difference.