【Agent-lightning】 - 1_环境搭建

源代码仓库: agent-lightning 源代码仓库

论文链接: Agent Lightning 论文

个人代码仓库: agent-lightning 个人仓库

Tutorial: Tutorial.md

记录一下使用 Agent-lightning 时的一些使用细节,这里主要记录一下最开始的环境搭建。

这里主要使用 uv工具 来搭建环境,步骤如下:

bash 复制代码
git clone https://github.com/Yhc-777/agent-lightning.git
cd agent-lightning

uv venv .venv --python 3.12
source .venv/bin/activate

uv sync
uv pip install -e .

# 可能在做项目时需要安装额外的库,例如:
uv pip install vllm==0.10.2
uv pip install verl==0.5.0
uv pip install torch==2.8.0 torchvision==0.23.0 --index-url https://download.pytorch.org/whl/cu128

mkdir pkgs && cd pkgs
wget https://github.com/Dao-AILab/flash-attention/releases/download/v2.8.1/flash_attn-2.8.1+cu12torch2.8cxx11abiTRUE-cp312-cp312-linux_x86_64.whl
uv pip install flash_attn-2.8.1+cu12torch2.8cxx11abiTRUE-cp312-cp312-linux_x86_64.whl

这个环境主要用于:

  • LLM的模型开发部署(vLLM)
  • DL的训练(PyTorch+CUDA,对版本有要求)
  • Attention计算(Flash Attention)
  • RL(Verl)

当然如果后面需要安装额外的库,或者兼容其他算法,也可以直接 uv pip install xxx 即可。

相关推荐
科技云报道2 分钟前
AI时代重新定义“信任”,瑞数信息发布全新AI安全产品体系
人工智能·安全
用户970161501687 分钟前
微服务里最危险的 DELETE,不是删不掉,是只删了一半
人工智能·后端
夏洛克信徒9 分钟前
当黄仁勋说出“AGI已来“:2026年9月大模型风暴观察
人工智能·gpt·chatgpt·agi
通问AI12 分钟前
用多模态图像模型批量生产电商主图:Prompt 模板化 + 自动化质检的工程实践
人工智能
也不知秋13 分钟前
从“能回答”到“能干活”:AI Agent真正落地,需要哪些工程能力?
人工智能·程序员
ACP广源盛1392462567314 分钟前
国产 PCIe4.0 交换芯片 IX8012@ACP:AI 推理服务器高速 IO 扩展方案解析
人工智能·硬件架构·pcie·国产芯片·ai服务器
2601_9620990821 分钟前
在Python中使用LSTM和PyTorch进行时间序列预测
pytorch·机器学习·lstm·时间序列预测·数据预处理
论文复现现场25 分钟前
单卡RTX 3090能训练,切到4卡却OOM:Accelerate多卡训练怎么排查?
人工智能·pytorch·云计算·gpu算力·多卡训练
2601_9563198826 分钟前
看到“2026年量化学习”时,交易想法要先拆成条件和动作
人工智能·python
武哥聊编程29 分钟前
【AI实战项目】AI Agent数据分析平台,基于SpringAI+Springboot+Vue+Agent的电商数据分析平台
人工智能·spring boot·数据分析·springai