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.
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.
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