github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-30 23:06:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

[workflow] fixed gpu memory check condition (#2659)

Browse Source
This commit is contained in:
Frank Lee 2023-02-10 09:59:07 +08:00 committed by GitHub
parent a255a38f7f
commit 94f87f9651
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
.github/workflows/build_on_schedule.yml vendored
View File

@ -23,7 +23,7 @@ jobs:
for i in $(seq 0 7);
do
gpu_used=$(nvidia-smi -i $i --query-gpu=memory.used --format=csv,noheader,nounits)
[ "$gpu_used" -le "10000" ] && avai=false
[ "$gpu_used" -gt "10000" ] && avai=false
done
echo "GPU is available: $avai"