RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cuda:1

[Mapzzone]

how to solve it!!!!!thanks!

[Nuzhny007]

Change self.device = torch.device('cuda') to self.device = torch.device('cuda:1')

[Mapzzone]

I did it,and then,the problem happened.....

------------------ 原始邮件 ------------------ 发件人: "wangguanan/HOReID" @.>; 发送时间: 2022年7月7日(星期四) 下午4:48 @.>; @.@.>; 主题: Re: [wangguanan/HOReID] RuntimeError: module must have its parameters and buffers on device cuda:0 (device_ids[0]) but found one of them on device: cuda:1 (Issue #28)

Change self.device = torch.device('cuda') to self.device = torch.device('cuda:1')

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you authored the thread.Message ID: @.***>