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.
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.