diff --git a/src/tcn/ci/pipeline/templates/gpu-mps-launcher.sh b/src/tcn/ci/pipeline/templates/gpu-mps-launcher.sh index 6a93f72..13855c0 100755 --- a/src/tcn/ci/pipeline/templates/gpu-mps-launcher.sh +++ b/src/tcn/ci/pipeline/templates/gpu-mps-launcher.sh @@ -68,7 +68,7 @@ else nvidia-cuda-mps-control -d fi # MPS server is socket base, leave time for the filesystem - sleep 10 + sleep 3 # Server should be spun, we restrict this rank to a single GPU GPU=$((LOCAL_RANK/PER_DEVICE_PROCESS)) export CUDA_VISIBLE_DEVICES=$GPU