diff --git a/utils/aws/resume.py b/utils/aws/resume.py
index 4b0d4246b594..e869834e96e7 100644
--- a/utils/aws/resume.py
+++ b/utils/aws/resume.py
@@ -28,7 +28,7 @@
 
     if ddp:  # multi-GPU
         port += 1
-        cmd = f'python -m torch.distributed.launch --nproc_per_node {nd} --master_port {port} train.py --resume {last}'
+        cmd = f'python -m torch.distributed.run --nproc_per_node {nd} --master_port {port} train.py --resume {last}'
     else:  # single-GPU
         cmd = f'python train.py --resume {last}'