ximinng / SVGDreamer

[CVPR 2024] Official implementation for "SVGDreamer: Text Guided SVG Generation with Diffusion Model" https://arxiv.org/abs/2312.16476
https://ximinng.github.io/SVGDreamer-project/
MIT License
218 stars 17 forks source link

cudaErrorInvalidDevice #19

Open davidaknowles opened 1 month ago

davidaknowles commented 1 month ago

Hi - the script successfully runs SD/DDPM but then hits an error:

pydiffvg/render_pytorch.py", line 692, in backward diffvg.render(scene,
RuntimeError: radix_sort: failed on 1st step: cudaErrorInvalidDevice: invalid device ordinal

Any ideas? This is on Ubuntu 24. torch finds the GPU fine (obviously because SD wouldn't run otherwise).

ximinng commented 1 month ago

Sorry, I have not encountered this problem. I think there are two possibilities: 1.diffvg is not installed correctly. 2. Test if pytorch can call the GPU correctly.