zcablii / LSKNet

(IJCV2024 & ICCV2023) LSKNet: A Foundation Lightweight Backbone for Remote Sensing
Other
489 stars 40 forks source link

RuntimeError: Unable to find a valid cuDNN algorithm to run convolution #67

Closed lsjEric closed 6 months ago

lsjEric commented 6 months ago

What's the feature?

I have two 4090 graphics cards. When I run the code with the first graphics card, there is no problem, but when I use the second graphics card, the above error appears. I would like to ask the following authors if they have encountered this problem.

Any other context?

No response

zcablii commented 6 months ago

Based on this answer, try a smaller model or a smaller batch size? Or you can find more solutions on Google, because this is likely not the issue of this repo.