diff --git a/README.md b/README.md index b543afaa..ecb87e2f 100644 --- a/README.md +++ b/README.md @@ -345,6 +345,8 @@ To enable FlashAttention-2 on the Windows platform, you need to install the prec
For Ascend NPU users +Join [NPU user group](assets/wechat_npu.jpg). + To utilize Ascend NPU devices for (distributed) training and inference, you need to install the **[torch-npu](https://gitee.com/ascend/pytorch)** library and the **[Ascend CANN Kernels](https://www.hiascend.com/developer/download/community/result?module=cann)**. | Requirement | Minimum | Recommend | diff --git a/README_zh.md b/README_zh.md index 2eb2cec0..3ef92697 100644 --- a/README_zh.md +++ b/README_zh.md @@ -345,6 +345,8 @@ pip install https://github.com/jllllll/bitsandbytes-windows-webui/releases/downl
昇腾 NPU 用户指南 +加入 [NPU 用户群](assets/wechat_npu.jpg)。 + 如果使用昇腾 NPU 设备进行(分布式)训练或推理,需要安装 **[torch-npu](https://gitee.com/ascend/pytorch)** 库和 **[Ascend CANN Kernels](https://www.hiascend.com/developer/download/community/result?module=cann)**。 | 依赖项 | 至少 | 推荐 | diff --git a/assets/wechat_npu.jpg b/assets/wechat_npu.jpg new file mode 100644 index 00000000..353e7603 Binary files /dev/null and b/assets/wechat_npu.jpg differ