drift-agentdojo-slim
收藏资源简介:
DRIFT AgentDojo Slim Docker Release 是一个用于运行 AgentDojo 评测任务的容器化发布包。它包含两个主要版本:CUDA 12 兼容版,由 vLLM 推理服务镜像(基于 CUDA 12.4、PyTorch 2.5.1)和 DRIFT + AgentDojo 客户端镜像组成,支持 Docker 和 nerdctl/containerd;以及原始 CUDA 13 一体化版。发布包提供自动检查、导入、启动、等待服务并运行评测的脚本(run-compatible.sh 或 run.sh),以及中文完整说明(QUICKSTART.md)和校验文件(checksums.sha256)。用户需自行提供模型权重(如 Llama-3.3-70B-Instruct),并确保宿主 NVIDIA 驱动兼容。该发布包适用于在支持 CUDA 的 GPU 环境中部署和运行 AgentDojo 安全评测基准。
DRIFT AgentDojo Slim Docker Release is a containerized release package for running AgentDojo evaluation tasks. It includes two main versions: a CUDA 12 compatible version consisting of a vLLM inference service image (based on CUDA 12.4, PyTorch 2.5.1) and a DRIFT + AgentDojo client image, supporting Docker and nerdctl/containerd; and a raw CUDA 13 all-in-one version. The package provides scripts for automatic checking, importing, starting, waiting for services, and running evaluations (run-compatible.sh or run.sh), along with a full Chinese guide (QUICKSTART.md) and checksum files (checksums.sha256). Users need to provide their own model weights (e.g., Llama-3.3-70B-Instruct) and ensure compatibility with the host NVIDIA driver. This release is suitable for deploying and running the AgentDojo security evaluation benchmark in CUDA-capable GPU environments.
DRIFT AgentDojo Slim Docker Release 数据集概述
基本信息
- 数据集名称:DRIFT AgentDojo Slim Docker Release
- 许可证:other
- 数据集类型:Docker 容器发布包(可迁移容器),不包含模型权重
版本与组成
1. 推荐版本:CUDA 12 兼容版
该版本将推理服务与评测客户端拆分为两个独立镜像,支持 Docker 和 nerdctl/containerd,包含以下组件:
| 组件 | 说明 |
|---|---|
drift-agentdojo-vllm-cu12.tar |
官方 vLLM 0.6.6.post1,PyTorch 2.5.1,CUDA 12.4 |
drift-agentdojo-client.tar |
DRIFT + AgentDojo 客户端 |
run-compatible.sh |
自动化脚本(检查、导入、启动、等待服务、运行评测) |
QUICKSTART.md |
中文完整说明文档 |
checksums.sha256 |
下载完整性校验文件 |
环境要求:
- 最低宿主 NVIDIA 驱动版本:525.60.13
- 容器内部 CUDA 版本与宿主驱动显示版本可不同(容器自带 CUDA 用户态库)
2. 原始版本:CUDA 13 一体化版
- 文件:
drift-agentdojo-slim.tar - 要求:NVIDIA 驱动 580 或更高
- 启动方式:使用
run.sh
已验证范围
在 4 × NVIDIA RTX A6000、驱动 580.173.02 环境下验证:
- 容器内 CUDA 可用并完成真实张量计算
- vLLM CUDA 12.4 成功加载真实 TinyLlama checkpoint
/v1/models和/v1/completions接口返回有效结果- DRIFT/AgentDojo 客户端可连接服务并完成补全请求
run-compatible.sh通过 Docker 和 nerdctl 2.3.5 + containerd 2.3.3 完整预检
运行要求(正式 LLM)
- 需自行提供 gated 权重(Llama-3.3-70B-Instruct,约 131 GiB)
- 总显存需求:约 150 GiB
- 详细说明见
QUICKSTART.md
使用方式示例
bash hf download Jinming2002/drift-agentdojo-slim drift-agentdojo-vllm-cu12.tar drift-agentdojo-client.tar run-compatible.sh QUICKSTART.md checksums.sha256 --repo-type dataset --local-dir . sha256sum -c checksums.sha256 --ignore-missing chmod +x run-compatible.sh WEIGHTS=/path/to/Llama-3.3-70B-Instruct ./run-compatible.sh
上游项目
- vLLM:https://github.com/vllm-project/vllm
- AgentDojo:https://github.com/ethz-spylab/agentdojo




