The PidiNet detector cannot run on any other GPU aside from GPU 0 due always initializing a specific tensor on the first available GPU.
Changing this line of code so that the tensor is created on the correct GPU fixes the issue and allows PidiNet to run on a device other than the primary GPU.
The PidiNet detector cannot run on any other GPU aside from GPU 0 due always initializing a specific tensor on the first available GPU.
Changing this line of code so that the tensor is created on the correct GPU fixes the issue and allows PidiNet to run on a device other than the primary GPU.