---
title: 执行PyTorch框架的训练任务时，提示找不到amp_C-昇腾社区
description: 开启watchdog功能后，下发PyTorch框架的训练任务，报错提示找不到amp_C。
keywords: amp_C;megatron_npu;PyTorch
url: https://www.hiascend.com/document/caselibrary/detail/mindxdl_0041
section: (其他)
---

# 执行PyTorch框架的训练任务时，提示找不到amp_C-昇腾社区

URL: https://www.hiascend.com/document/caselibrary/detail/mindxdl_0041
描述: 开启watchdog功能后，下发PyTorch框架的训练任务，报错提示找不到amp_C。
关键词: amp_C;megatron_npu;PyTorch

昇腾文档 [了解详情](https://www.hiascend.com/document)

故障案例 [了解详情](https://www.hiascend.com/document/caselibrary)

执行PyTorch框架的训练任务时，提示找不到amp_C

执行PyTorch框架的训练任务时，提示找不到amp_C

2025/01/26

896

问题信息

| 问题来源 | 产品大类 | 产品子类 | 关键字 |
| --- | --- | --- | ---|
| 官方 | 集群调度 | 断点续训 | amp_C、megatron_npu、PyTorch |

#### 问题现象描述

开启watchdog功能后，下发PyTorch框架的训练任务，报错提示找不到amp_C。

#### 原因分析

镜像中找不到megatron_npu路径。

#### 解决措施

在train_start.sh中新增以下加粗的环境变量，设置megatron_npu在Python中的环境变量。

```
...
# env for breakpoint ckpt
export RESUME_MODE_ENABLE=1

export PYTHONPATH=$PYTHONPATH:MEGATRON_LM的路径/megatron_npu
```

本页内容

- 问题信息
- 问题现象描述
- 原因分析
- 解决措施