yangxy / GPEN

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

cuda:1 and the default cuda:0 give different results?

e4s2022 opened this issue · comments

It is a bit wired that when I changed the default cuda to cuda:1(see here), the enhanced face turns to be an image with full noise.

image

I can confirm the default cuda works normally.

UPDATE

import torch 
torch.cuda.set_device(1)

The above two lines work well if one wants to use a different card (with the default cuda)...but...still don't know why cuda:1 fails.