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
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
False, and any attempt to
x.to(device) returns initialization errors.
Restart of python kernel doesn’t help the situation.