    main() Once you’ve completed the inference script, use the --nproc_per_node argument to specify the number of GPUs to use and call torchrun to run the script:   Copied torchrun run_distributed.py --nproc_per_node=2