gpu_lock:使用基于文件的锁定在Rudolf上保留GPU

GPU锁 此软件包可管理用户之间的GPU访问，以防止我们意外地在另一个用户当前用于其研究的GPU上启动进程。 这种方法优于使用nvidia-smi，因为即使在脚本的短暂中断（例如，在运行超参数优化时查找新的超参数集）期间不使用GPU，GPU也会保持锁定状态。 该软件包通过将文件放在/ var / tmp / gpu_locks中来管理用户之间的GPU访问。 它会自动将CUDA_VISIBLE_DEVICES设置为您持有锁定的GPU的uid。 所有锁定均基于共识-仅在每个人都使用此库的情况下才起作用。 拥有锁并不会系统地阻止其他用户使用您“锁定”的GPU。 安装： 您可以使用pip安装此软件包： pip install git+https://github.com/idsc-frazzoli/gpu_lock.git 更新中 由于此程序包仍在开发中，因此我们有时可能会将更改推送到此仓库