#gpu #api-key #lambda #instance #cloud #ssh-key #interact

app lambda_cli

A Rust应用程序用于与Lambda Cloud GPU API交互

1 个不稳定版本

0.1.0 2024年5月27日

#1887 in Web编程

MIT许可证

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