Open hxngiee opened 3 years ago
Looks like you are using more than one GPU. A possible solution is to change the device from 'cuda' to a specific GPU device, e.g. 'cuda:0'.
https://github.com/FuxiCV/MeInGame/blob/da9cff0251df9cf0378f4e171f55bbd0ed22b04b/main.py#L95
Is there a way to run the code on multiple GPUs as my single GPU has low memory?
Thanks for sharing your great work
I got some error when run the code
I'd appreciate your advice