github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

pull/2665/head
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"