diff --git a/debug/accuracy_tools/kj600/README.md b/debug/accuracy_tools/kj600/README.md index f85f1fda4ed4416f17dfaa8a10b517e4c15eb5a1..8fa4824e11c40730dbe1fde4c811bb2f84108a36 100644 --- a/debug/accuracy_tools/kj600/README.md +++ b/debug/accuracy_tools/kj600/README.md @@ -8,13 +8,18 @@ ### 1. 安装依赖 -| 依赖软件 | -|-------------| -| torch | -| torch_npu | -| torchvision | -| tensorboard | -| matplotlib | +| 依赖软件 | 版本 | +|------------|-----------| +| torch | 1.8.0+ | +| torch_npu | 适配torch版本 | +| pandas | 1.3.5 | +| pymysql | 1.1.1 | +| matplotlib | 3.5.2 | + +| 适配框架 | 版本 | +|-----------|---------| +| megatron | 0.6.0 | +| deepspeed | 0.12.6 | ### 2. 安装 kj600