3825e99ee36bd237d11f702af93a2acba4496c9b,utils/torch_utils.py,,select_device,#Any#,10

Before Change

        device = torch.device("cuda:0" if cuda else "cpu")

        if torch.cuda.device_count() > 1:
            device = torch.device("cuda" if cuda else "cpu")
            print("Found %g GPUs" % torch.cuda.device_count())
            // print("Multi-GPU Issue: https://github.com/ultralytics/yolov3/issues/21")
            // torch.cuda.set_device(0)  // OPTIONAL: Set your GPU if multiple available
            // print("Using ", torch.cuda.device_count(), " GPUs")

After Change

              (x[0].name, x[0].total_memory / c))
        if ng > 0:
            // torch.cuda.set_device(0)  // OPTIONAL: Set GPU ID
            for i in range(1, ng):
                print("           device%g _CudaDeviceProperties(name="%s", total_memory=%dMB)" %
                      (i, x[i].name, x[i].total_memory / c))

    return device

Instances