Not really. The code would need a decent amount of change to not use at least 2 GPUs. It also has some C++ TorchScript module that might be part of the issue too. Is there another way to debug the issue?