Traceback (most recent call last):

  File "/home/wz/pytorch_main.py", line 516, in <module>

    main()

  File "/home/wz/pytorch_main.py", line 126, in main

    main_worker(args.gpu, ngpus_per_node, args)

  File "/home/wz/pytorch_main.py", line 185, in main_worker

    model = torch.nn.DataParallel(model).cuda()

            ^^^^^^^^^^^^^^^^^^^^^^^^^^^^

  File "/usr/local/python3.11.10/lib/python3.11/site-packages/torch/nn/parallel/data_parallel.py", line 165, in __init__

    _check_balance(self.device_ids)

  File "/usr/local/python3.11.10/lib/python3.11/site-packages/torch/nn/parallel/data_parallel.py", line 45, in _check_balance

    if warn_imbalance(lambda props: props.multi_processor_count):

       ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

  File "/usr/local/python3.11.10/lib/python3.11/site-packages/torch/nn/parallel/data_parallel.py", line 33, in warn_imbalance

    values = [get_prop(props) for props in dev_props]

             ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

  File "/usr/local/python3.11.10/lib/python3.11/site-packages/torch/nn/parallel/data_parallel.py", line 33, in <listcomp>

    values = [get_prop(props) for props in dev_props]

              ^^^^^^^^^^^^^^^

  File "/usr/local/python3.11.10/lib/python3.11/site-packages/torch/nn/parallel/data_parallel.py", line 45, in <lambda>

    if warn_imbalance(lambda props: props.multi_processor_count):

                                    ^^^^^^^^^^^^^^^^^^^^^^^^^^^

AttributeError: 'torch_npu._C._NPUDeviceProperties' object has no attribute 'multi_processor_count'