ModuleNotFoundError: No module named ‘apex’

ModuleNotFoundError: No module named 'apex' 表示 Python 在搜索模块时无法找到名为 apex 的模块。这通常是因为您没有安装 apex 模块或安装不正确。

apex 是一个针对混合精度训练和优化的 PyTorch 扩展库，您可以通过以下方式安装它：

pip install -U git+https://github.com/NVIDIA/apex

请注意，安装 apex 需要满足一些依赖关系，例如 CUDA 和 CuDNN，因此请确保您的环境中已经安装了这些依赖项。

如果您已经安装了 apex，但仍然遇到此错误，请确保您的 Python 环境中的 PYTHONPATH 环境变量包含指向 apex 的路径。例如，如果您在 ~/apex 中安装了 apex，则可以将以下行添加到您的 .bashrc 或 .bash_profile 文件中:

export PYTHONPATH=$PYTHONPATH:~/apex

然后，重启终端会话或运行 source ~/.bashrc 或 source ~/.bash_profile 以应用更改。

方法二

当运行 import apex 时报错 ModuleNotFoundError: No module named apex。这是因为没有安装 nvidia 的apex库 简介。安装方法(直接在当前目录执行即可，安装后可再删除文件夹)：

git clone https://www.github.com/nvidia/apex
cd apex
python setup.py install [--cuda_ext] [--cpp_ext]

利用源码进行安装，千万不要图方便：

pip install apex

否则装上了也会出现很多缺少依赖的错误,采用以下方式进行卸载

pip uninstall apex

错误2：error: legacy-install-failure

通常与 pip 版本不兼容有关，建议您先尝试升级 pip 的版本，可以使用以下命令：

pip install --upgrade pip

然后在通过源码进行安装，进入到apex文件中

python setup.py install --cpp_ext --cuda_ext

在执行上述命令之前，请确保已经按照官方文档中的说明正确地配置了 CUDA 和 PyTorch。