RuntimeError: CUDA error: no kernel image is available for execution on the device

JCruan519 / VM-UNet

(ARXIV24) This is the official code repository for "VM-UNet: Vision Mamba UNet for Medical Image Segmentation".

Apache License 2.0

473 stars 22 forks source link

Open 050603 opened 7 months ago

050603 commented 7 months ago

各位大佬，这个问题有没有遇到过？我看csdn链接这篇文章的方法是降低torch版本，但我总觉得不太对劲。有没有其他方法呢？

JCruan519 commented 6 months ago

@050603 Hello, I suggest you can directly find and install the corresponding CUDA version of PyTorch from this link (https://download.pytorch.org/whl/torch_stable.html).