How to change device of gpu

clovaai / stargan-v2

StarGAN v2 - Official PyTorch Implementation (CVPR 2020)

Other

3.48k stars 653 forks source link

How to change device of gpu #158

Open sevenPot7 opened 1 year ago

sevenPot7 commented 1 year ago

I used "device = torch.device('cuda:1') if torch.cuda.is_available() else torch.device('cpu') " but it's not work. Hope somebody can give me advice, thanks.

SubeBila commented 2 months ago

Did you test if GPU0 is available?