pllm2-full-dump / gpu_utils /utils_readme.md
yutong.song
Full PLLM2.0 dump
4b1637c
|
Raw History Blame Contribute Delete
613 Bytes
# 直接在PLLM1.0的目录下运行:
python3 gpu_utils/gpu_stress.py --duration 2.0 --target_utilization 0.8
python3 gpu_utils/gpu_stress.py --info_only
# 使用所有可用GPU,运行30分钟,目标利用率50%
python3 gpu_utils/gpu_stress.py --duration 0.5 --target_utilization 0.5
# 指定使用2个GPU,运行10分钟,目标利用率80%
cd /Users/bytedance/Desktop/PLLM1.0-Sep23 && python3 gpu_utils/gpu_stress.py --duration 0.1667 --target_utilization 0.8 --num_gpus 2
gpu_utils/python3 clean_gpu.py
# 重复清理3次,每次间隔2秒
gpu_utils/python3 clean_gpu.py --repeat 3 --interval 2