文件名称：simple_gpu_scheduler:用于在GPU上运行作业的简单计划程序
文件大小：19KB
文件格式：ZIP
更新时间：2024-03-28 14:55:21
Python
simple_gpu_scheduler 一个简单的调度程序，可以在各个GPU上运行命令。 遵循，此脚本仅通过stdin接受命令，并通过设置CUDA_VISIBLE_DEVICES变量在特定的GPU上执行命令。 使用登录外壳执行读取的命令，因此重定向>管道| 以及所有其他种类的贝壳魔法都可以使用。 安装 该软件包可以简单地从安装 $ pip3 install simple-gpu-scheduler 简单的例子 假设您有一个文件gpu_commands.txt ，其中包含要在GPU 0、1和2上并行执行的命令： $ cat gpu_commands.txt python train_model.py --lr 0.001 --output run_1 python train_model.py --lr 0.0005 --output run_2 python train_model
【文件预览】：
simple_gpu_scheduler-master
----setup.py(966B)
----.gitignore(1KB)
----Makefile(392B)
----MANIFEST.in(34B)
----LICENSE(34KB)
----README.md(3KB)
----simple_gpu_scheduler()
--------hypersearch.py(4KB)
--------scheduler.py(3KB)
--------__init__.py(0B)