Skip to content

godweiyang/GrabGPU

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

31 Commits
 
 
 
 
 
 

Repository files navigation

抢占显卡脚本

知乎介绍

下载方法(向下兼容)

CUDA 10.1:

wget https://github.com/godweiyang/GrabGPU/releases/download/v1.0.1/gg_cu101

CUDA 11.0:

wget https://github.com/godweiyang/GrabGPU/releases/download/v1.0.1/gg_cu110

CUDA 12.1:

wget https://github.com/godweiyang/GrabGPU/releases/download/v1.0.1/gg_cu121

如果你的 CUDA 版本不适配,请自行编译

nvcc gg.cu -o gg

抢占到显卡后自动执行默认脚本

使用方法:

./gg <占用显存 (GB)> <占用时间 (h)> <显卡序号>

举例:
抢占 16 GB 显存 24 小时,使用 GPU 0, 1, 2, 3 来运行默认脚本。

./gg 16 24 0,1,2,3

抢占到显卡后自动执行自定义程序(比如训练模型)

使用方法:

./gg <占用显存 (GB)> <占用时间 (h)> <显卡序号> <自定义脚本路径(.sh文件)>

举例:
抢占 16 GB 显存 24 小时,使用 GPU 0, 1, 2, 3 来运行自定义脚本 run.sh。注意这里的占用时间是无效的,会直到自定义脚本执行完毕才释放显卡。

./gg 16 24 0,1,2,3 run.sh

About

一款便捷的抢占显卡脚本

Resources

Stars

Watchers

Forks

Packages

No packages published

Languages