But I’m moving tensor to the specific device.
Moreover, if I set the only one visible GPU, all is ok with 0 or 1, but doesn’t work for 3. (2 is the dropped one).
And yes, I’ve checked the device id’s, and moreover I’ve used code like this before GPU failed. So everything is ok with id, I think.