Skip to content

Commit

Permalink
Lower sleep for MPS start to reasonnable 3s
Browse files Browse the repository at this point in the history
  • Loading branch information
FlorianDeconinck committed May 8, 2024
1 parent 731d104 commit 68a4803
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/tcn/ci/pipeline/templates/gpu-mps-launcher.sh
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit 68a4803

Please sign in to comment.