from apex import amp正确安装apex

在Linux环境下安装apex

首先创建合适python版本的anaconda环境

$ git clone https://github.com/NVIDIA/apex
$ cd apex
$ pip install packaging
$ python setup.py install

安装成功后退出apex当前路径

$ cd ..

不要使用以下命令进行安装

$ pip install apex

 如果使用了该方法需要先卸载再安装

$ pip uninstall apex

复制apex进入apex对应目录后建议直接运行setup.py，其他方法可能报错导致安装失败。

运行代码如果显示apex相关报错，可考虑调整网络结构中出现的各种小问题~不需要换为cuda.amp的形式计算损失及优化