如何安装:
首先 pip uninstall apex卸载、apex文件夹也删除,
重新安装,具体安装步骤如下:
git clone https://www.github.com/nvidia/apex
cd apex
python setup.py install --cuda_ext --cpp_ext
只要出现了下面的语句就是完成安装了:
可能的错误:
错误一:
ModuleNotFoundError: No module named ‘amp_C‘
直接报错的那个import注释掉即可,也就是注释下面这一行:
import amp_C
错误二:
ImportError: cannot import name ‘container_abcs‘ from ‘torch._six‘
解决方法:
ImportError: cannot import name ‘container_abcs‘ from ‘torch._six‘_importerror: cannot import name 'container_abcs_懵懂的海哥·的博客-CSDN博客
其他错误:
apex安装常见的三个报错并成功解决(亲测有效)_pip install apex_weixin_59726951的博客-CSDN博客