Hello!
I have 4 GPU on my computer. And I have one lost:
$>nvidia-smi
Unable to determine the device handle for GPU 0000:86:00.0: GPU is lost.

But another one is still working. e.g.
$>nvidia-smi -i 0

works good.
For the some reasons, I don’t want to reboot (some other computations).
Is there any way to specify which GPU to use for the PyTorch?
Right now even torch.cuda.is_available() returns False, and any attempt to x.to(device) returns initialization errors.
Restart of python kernel doesn’t help the situation.