当前采用多GPU训练报错：/home/heyuting/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/distributed/launch.py:181: FutureWarning: The module torch.distributed.launch is deprecated
and will be removed in future. Use torchrun.
Note that --use-env is set by default in torchrun.
If your script expects `--local-rank` argument to be set, please
change it to read from `os.environ['LOCAL_RANK']` instead. See 
https://pytorch.org/docs/stable/distributed.html#launch-utility for 
further instructions

  warnings.warn(
[2024-06-14 21:36:21,506] torch.distributed.run: [WARNING] 
[2024-06-14 21:36:21,506] torch.distributed.run: [WARNING] *****************************************
[2024-06-14 21:36:21,506] torch.distributed.run: [WARNING] Setting OMP_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed. 
[2024-06-14 21:36:21,506] torch.distributed.run: [WARNING] *****************************************
2024-06-14 21:36:23.475564: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 AVX512F FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2024-06-14 21:36:23.503752: I tensorflow/core/platform/cpu_feature_guard.cc:210] This TensorFlow binary is optimized to use available CPU instructions in performance-critical operations.
To enable the following instructions: AVX2 AVX512F FMA, in other operations, rebuild TensorFlow with the appropriate compiler flags.
2024-06-14 21:36:24.309452: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
2024-06-14 21:36:24.358850: W tensorflow/compiler/tf2tensorrt/utils/py_utils.cc:38] TF-TRT Warning: Could not find TensorRT
usage: run_training.py [-h] --script SCRIPT --config CONFIG [--cudnn_benchmark CUDNN_BENCHMARK]
                       [--local_rank LOCAL_RANK] [--save_dir SAVE_DIR] [--seed SEED]
                       [--use_lmdb {0,1}] [--script_prv SCRIPT_PRV] [--config_prv CONFIG_PRV]
                       [--use_wandb {0,1}] [--distill {0,1}] [--script_teacher SCRIPT_TEACHER]
                       [--config_teacher CONFIG_TEACHER]
run_training.py: error: unrecognized arguments: --local-rank=1
usage: run_training.py [-h] --script SCRIPT --config CONFIG [--cudnn_benchmark CUDNN_BENCHMARK]
                       [--local_rank LOCAL_RANK] [--save_dir SAVE_DIR] [--seed SEED]
                       [--use_lmdb {0,1}] [--script_prv SCRIPT_PRV] [--config_prv CONFIG_PRV]
                       [--use_wandb {0,1}] [--distill {0,1}] [--script_teacher SCRIPT_TEACHER]
                       [--config_teacher CONFIG_TEACHER]
run_training.py: error: unrecognized arguments: --local-rank=0
[2024-06-14 21:36:26,516] torch.distributed.elastic.multiprocessing.api: [ERROR] failed (exitcode: 2) local_rank: 0 (pid: 451104) of binary: /home/heyuting/anaconda3/envs/mamba/bin/python
Traceback (most recent call last):
  File "/home/heyuting/anaconda3/envs/mamba/lib/python3.10/runpy.py", line 196, in _run_module_as_main
    return _run_code(code, main_globals, None,
  File "/home/heyuting/anaconda3/envs/mamba/lib/python3.10/runpy.py", line 86, in _run_code
    exec(code, run_globals)
  File "/home/heyuting/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/distributed/launch.py", line 196, in <module>
    main()
  File "/home/heyuting/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/distributed/launch.py", line 192, in main
    launch(args)
  File "/home/heyuting/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/distributed/launch.py", line 177, in launch
    run(args)
  File "/home/heyuting/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/distributed/run.py", line 797, in run
    elastic_launch(
  File "/home/heyuting/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 134, in __call__
    return launch_agent(self._config, self._entrypoint, list(args))
  File "/home/heyuting/anaconda3/envs/mamba/lib/python3.10/site-packages/torch/distributed/launcher/api.py", line 264, in launch_agent
    raise ChildFailedError(
torch.distributed.elastic.multiprocessing.errors.ChildFailedError: 
============================================================
lib/train/run_training.py FAILED
------------------------------------------------------------
Failures:
[1]:
  time      : 2024-06-14_21:36:26
  host      : server12
  rank      : 1 (local_rank: 1)
  exitcode  : 2 (pid: 451105)
  error_file: <N/A>
  traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
------------------------------------------------------------
Root Cause (first observed failure):
[0]:
  time      : 2024-06-14_21:36:26
  host      : server12
  rank      : 0 (local_rank: 0)
  exitcode  : 2 (pid: 451104)
  error_file: <N/A>
  traceback : To enable traceback see: https://pytorch.org/docs/stable/elastic/errors.html
============================================================