search

hkchengrex / XMem

[ECCV 2022] XMem: Long-Term Video Object Segmentation with an Atkinson-Shiffrin Memory Model
https://hkchengrex.com/XMem/
MIT License
1.76k stars 192 forks source link

Can I train in Windows enviroment? #119

Closed returnlsz closed 1 year ago

returnlsz commented 1 year ago

I notice that you test your code in Ubuntu,but I have built it in Windows. So when I start training it cause the error: RuntimeError: Distributed package doesn't have NCCL built in Windows doesnt support nccl,but if I modify it to gloo can it work? import os os.environ["PL_TORCH_DISTRIBUTED_BACKEND"] = "gloo" I will be very appreciated if you could answer and solve it.

hkchengrex commented 1 year ago

If PyTorch does not complain, it should be fine.

returnlsz commented 1 year ago

If PyTorch does not complain, it should be fine.

It seems Windows still doesnt support.I wonder what Ubuntu Version is suit for this work

hkchengrex commented 1 year ago

All recent ones should work.