apex安装、解决ModuleNotFoundError: No module named ‘amp_C‘报错

最新推荐文章于 2024-01-18 15:17:51 发布

是七叔呀

最新推荐文章于 2024-01-18 15:17:51 发布

阅读量1.3w

点赞数 6

本文链接：https://blog.csdn.net/m0_46378271/article/details/121017309

版权

深度学习专栏收录该内容

53 篇文章

订阅专栏

本文详细描述了在安装Apex过程中遇到的ModuleNotFoundError: 'amp_C'问题，包括官方教程和社区解决方案。重点在于如何修复与torch版本不匹配导致的错误，并提供了一步步的安装和调试指南。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

一、apex安装

1. 官网的Quick Start，但是安装在第三步会报错：

2. 知乎，第四步也会报错：

3. 然后卸载apex（第三步不同）按照以下方式重新安装成功：

二、解决ModuleNotFoundError: No module named 'amp_C'报错

一、apex安装

1. 官网的Quick Start，但是安装在第三步会报错：

官网：GitHub - NVIDIA/apex: A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
 
报错是因为基础环境有误，参考
https://blog.csdn.net/justforacm/article/details/115428583

2. 知乎，第四步也会报错：

git clone https://github.com/NVIDIA/apex
cd apex
python3 setup.py install
python setup.py install --cuda_ext --cpp_ext

3. 然后卸载apex（第三步不同）按照以下方式重新安装成功：

pip uninstall apex
git clone https://www.github.com/nvidia/apex
cd apex
python setup.py install

二、解决ModuleNotFoundError: No module named 'amp_C'报错

之后开始运行如下代码训练UNet+++的时候：

python train.py -g 0 -u v3 -e 200 -b 1 -l 0.1 -s 0.5 -v 15.0

会报错amp_C，刚开始是认为没有amp_C 这个模块，pip install也没有，上网找了好久也没搞定。

后面发现这个没太大关系，我的报错只是因为torch版本和GPU不匹配，但是注释掉最后报错的行之后，代码就能跑通了：