1 个不稳定版本
0.1.0 | 2024年5月27日 |
---|
#1887 in Web编程
380KB
282 行
Lambda CLI
Lambda CLI是用于与Lambda Labs云GPU API交互的命令行工具。
功能
- 验证API密钥
- 列出所有可用的GPU实例
- 使用指定的SSH密钥启动GPU实例
- 停止指定的GPU实例
- 列出所有正在运行的GPU实例
- 当GPU实例可用时持续查找并启动
安装
要使用Lambda CLI,您需要安装Rust和Cargo。您可以通过以下说明安装Rust和Cargo:这里。
克隆存储库并导航到项目目录
git clone https://github.com/cybrly/lambda_cli.git
cd lambda_cli
cargo build --release
用法
在开始使用Lambda CLI之前,请将Lambda API密钥设置为环境变量
export LAMBDA_API_KEY=your_api_key
运行CLI工具
./target/release/lambda_cli [COMMAND]
命令
- list: 列出所有可用的GPU实例。
- start --gpu <GPU_TYPE> --ssh <SSH_KEY>: 使用指定的SSH密钥启动GPU实例。
- stop --gpu <GPU_INSTANCE_ID>: 停止指定的GPU实例。
- running: 列出所有正在运行的GPU实例。
- find --gpu <GPU_TYPE> [--ssh <SSH_KEY>] [--sec ]: 当GPU实例可用时持续查找并启动。
示例
验证API密钥
./target/release/lambda_cli
列出所有可用的GPU实例
./target/release/lambda_cli list
使用指定的SSH密钥启动GPU实例
./target/release/lambda_cli start --gpu "gpu_1x_a10" --ssh "Chris"
停止指定的GPU实例
./target/release/lambda_cli stop --gpu "instance_id"
列出所有正在运行的GPU实例
./target/release/lambda_cli running
当GPU实例可用时持续查找并启动
./target/release/lambda_cli find --gpu "8x_h100" --ssh "SSH_KEY_NAME" --sec 30
依赖关系
~9–21MB
~306K SLoC