huuuuusy/videocube-toolkit

torch.device('cuda:2,3')

MzeroMiko opened this issue · 1 comments

MzeroMiko commented

In my device, it is not allowed to use torch.device('cuda:2,3') (in siamfc.py: TrackerSiamFC.init).
We recommend to delete this or use nn.DataParallel or torch.distribute etc instead.

huuuuusy commented

Thanks a lot; I have updated the codes based on your suggestion.