stargan-v2 icon indicating copy to clipboard operation
stargan-v2 copied to clipboard

Published 20 hours ago • verified publisher icon clovaai

How to change device of gpu

Open sevenPot7 opened this issue 1 year ago • 1 comments

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.

sevenPot7 avatar Jun 14 '23 12:06 sevenPot7