src/transformers/benchmark/benchmark_args.py/PyTorchBenchmarkArguments/n_gpu
class PyTorchBenchmarkArguments:    def n_gpu(self):
        requires_backends(self, ["torch"])
        return self._setup_devices[1]