·torch_npu

Guía de capacidad, implementación y verificación del entorno de Huawei Ascend Extension para PyTorch (torch_npu). Se aplica automáticamente cuando los usuarios usan @torch_npu, Ascend NPU, CANN o necesitan migrar PyTorch a NPU; cuando los usuarios usan @torch_npu_doc, proporciona una descripción de la capacidad de la documentación china dentro del proyecto basada en la documentación de referencia de habilidades.

14Instalaciones·1Tendencia·@ascend-ai-coding

Instalación

$npx skills add https://github.com/ascend-ai-coding/awesome-ascend-skills --skill torch_npu

Cómo instalar torch_npu

Instala rápidamente el skill de IA torch_npu en tu entorno de desarrollo mediante línea de comandos

  1. Abrir Terminal: Abre tu terminal o herramienta de línea de comandos (Terminal, iTerm, Windows Terminal, etc.)
  2. Ejecutar comando de instalación: Copia y ejecuta este comando: npx skills add https://github.com/ascend-ai-coding/awesome-ascend-skills --skill torch_npu
  3. Verificar instalación: Una vez instalado, el skill se configurará automáticamente en tu entorno de programación con IA y estará listo para usar en Claude Code, Cursor u OpenClaw

Fuente: ascend-ai-coding/awesome-ascend-skills.

SKILL.md

Ver original

检查 PyTorch 与 Python 版本是否在配套范围内(参见 README.zh.md 中的「PyTorch与Python版本配套表」):

| 设备与内存 | torch.npu:设备管理、devicecount、currentdevice、setdevice、synchronize、Stream/Event、内存统计与分配(memoryallocated、emptycache、MemPool 等)。 | | 张量/存储 | tensor.npu()、tensor.isnpu、NPU Storage、序列化 torch.save/load 支持 NPU,DDP/多进程 reductions。 | | 训练/优化 | torch.npu.amp 混合精度、torchnpu.optim、FSDP 补丁(ShardedGradScaler)、梯度检查点默认 NPU。 |

| 分布式 | torchnpu.distributed:HCCL/LCCL 后端、ishcclavailable、reinitprocessgroup、RPC、symmetric memory、DTensor 规则。 | | 扩展 API | torchnpu.contrib:NMS、IoU 系列、ROIAlign、DCN、FusedAttention、自定义模块(如 DropoutWithByteMask)等。 | | 图与编译 | NPU Graph(npugraphify)、Dynamo、Inductor、torch.compile 支持 NPU。 |

Guía de capacidad, implementación y verificación del entorno de Huawei Ascend Extension para PyTorch (torch_npu). Se aplica automáticamente cuando los usuarios usan @torch_npu, Ascend NPU, CANN o necesitan migrar PyTorch a NPU; cuando los usuarios usan @torch_npu_doc, proporciona una descripción de la capacidad de la documentación china dentro del proyecto basada en la documentación de referencia de habilidades. Fuente: ascend-ai-coding/awesome-ascend-skills.

Datos (listos para citar)

Campos y comandos estables para citas de IA/búsqueda.

Comando de instalación
npx skills add https://github.com/ascend-ai-coding/awesome-ascend-skills --skill torch_npu
Categoría
</>Desarrollo
Verificado
Primera vez visto
2026-03-09
Actualizado
2026-03-10

Browse more skills from ascend-ai-coding/awesome-ascend-skills

Respuestas rápidas

¿Qué es torch_npu?

Guía de capacidad, implementación y verificación del entorno de Huawei Ascend Extension para PyTorch (torch_npu). Se aplica automáticamente cuando los usuarios usan @torch_npu, Ascend NPU, CANN o necesitan migrar PyTorch a NPU; cuando los usuarios usan @torch_npu_doc, proporciona una descripción de la capacidad de la documentación china dentro del proyecto basada en la documentación de referencia de habilidades. Fuente: ascend-ai-coding/awesome-ascend-skills.

¿Cómo instalo torch_npu?

Abre tu terminal o herramienta de línea de comandos (Terminal, iTerm, Windows Terminal, etc.) Copia y ejecuta este comando: npx skills add https://github.com/ascend-ai-coding/awesome-ascend-skills --skill torch_npu Una vez instalado, el skill se configurará automáticamente en tu entorno de programación con IA y estará listo para usar en Claude Code, Cursor u OpenClaw

¿Dónde está el repositorio de origen?

https://github.com/ascend-ai-coding/awesome-ascend-skills