https://github.com/wilmerwang/gpusnatcher

抢占显卡
https://github.com/wilmerwang/gpusnatcher

pytorch tensorflow

Last synced: 3 days ago
JSON representation

抢占显卡

• Host: GitHub
• URL: https://github.com/wilmerwang/gpusnatcher
• Owner: wilmerwang
• License: mit
• Created: 2020-09-18T01:12:34.000Z (over 4 years ago)
• Default Branch: master
• Last Pushed: 2024-10-10T10:58:51.000Z (4 months ago)
• Last Synced: 2025-01-17T21:32:57.492Z (11 days ago)
• Topics: pytorch, tensorflow
• Language: Python
• Homepage:
• Size: 35.2 KB
• Stars: 62
• Watchers: 1
• Forks: 5
• Open Issues: 3
• Metadata Files:
  • Readme: README.md
  • License: LICENSE

Awesome Lists containing this project

README

        
# GPUSnatcher

GPUSnatcher是用来在抢占显卡的脚本，主要是在实验室显卡资源紧张且自己亟需使用显卡的情况下，使用该脚本自动抢占一个或多个显卡。

**建议将scramble4gpu.py更改为train.py，并设置仅自己可见，防止被打...**

**请勿恶意抢占!!**

## 依赖

- numpy

- torch or tensorflow

## 使用

- **请先配置Email，否则当抢占GPU之后，不能发送Email**

```

git clone https://github.com/wilmerwang/GPUSnatcher.git

cd GPUSnatcher

# ./email_conf.json

# 建议使用qq邮箱服务,如果用的其他邮箱服务器，请自行设置

{

  "host": "smtp.qq.com",  # qq邮箱server

  "user": "[email protected]",  # 要登陆的qq账号

  "pwd": "xxxxxxxxxxxxxxxx",  # SMTP授权码, qq邮箱--> 设置 --> 账号 --> IMAP/SMTP服务开启 --> 生成授权码

  "sender": "[email protected]",  # 发送者

  "receiver": "[email protected]"  # 接收邮箱,可以是列表比如["[email protected]", "[email protected]"]

}

```

- 配置之后运行程序

```shell

python scramble4gpu.py

```

### 可选参数

- -p --proportion 显卡空闲内存 / 全部内存 的阈值，取值在0-1之间。当p取1的时候，表示仅仅列出完全没有被使用的显卡。默认为0.8。

- -n --gpu_nums 需要抢占的GPU数量，建议不要抢太多，容易挨揍。默认是1。

- -t --times 抢占显卡之后，自动释放显卡的时间。默认是30分钟。

- -e --email_conf email的配置参数，默认在./email_conf.json

当想自己设置以上参数的时候：

```shell

# 查看参数详情

python scramble4gpu.py -h

# 查找Free显存大于0.9的显卡，抢占4个，1800秒后自动释放,email配置路径为./email_conf.json

python scramble4gpu.py -p 0.9 -n 4 -t 1800 -e ./email_conf.json

```