why my using device is cpu?

hkchengrex / Cutie

[CVPR 2024 Highlight] Putting the Object Back Into Video Object Segmentation

https://hkchengrex.com/Cutie/

MIT License

579 stars 60 forks source link

why my using device is cpu? #81

Closed nirvanaYu777 closed 4 days ago

nirvanaYu777 commented 1 week ago

I've got NVidia 4090 and setup all the requirements include cuda,pytorch etc.
My system is windows 11. This is the message: E:\XMem2\Cutie\Cutie>python interactive_demo.py --video ./examples/6.mp4 --num_objects 3 Using device: cpu Single object: False Workspace is in: ./workspace\6 5397 images found.

hkchengrex commented 1 week ago

What do you get when you run python -c "import torch; print(torch.cuda.is_available())"?

hkchengrex commented 4 days ago

Please feel free to re-open if there are any updates.