ai-for-science-oligoformer
Installation
SKILL.md
OligoFormer 昇腾 NPU 迁移 Skill
本 Skill 提供将 OligoFormer(siRNA 效能预测模型)从 CUDA 迁移到昇腾 NPU 的完整步骤。 项目地址:https://github.com/lulab/OligoFormer.git
前置条件
| 项目 | 要求 |
|---|---|
| 硬件 | Ascend910 系列(至少 1 卡) |
| OS | openEuler / Ubuntu / KylinOS(aarch64 或 x86_64) |
| CANN | ≥ 8.0(推荐 8.2+ 或 8.3.RC1) |
| Python | 3.10(原项目要求 3.8,因 torch 2.5.1 不支持 3.8,升级至 3.10) |
| PyTorch | 2.5.1 |
| torch_npu | 2.5.1 |
| conda | 已安装 |