ModuleNotFoundError: No module named ‘a(chǎn)pex’

ModuleNotFoundError: No module named 'apex' 表示 Python 在搜索模塊時(shí)無(wú)法找到名為 apex 的模塊。這通常是因?yàn)槟鷽]有安裝 apex 模塊或安裝不正確。

apex 是一個(gè)針對(duì)混合精度訓(xùn)練和優(yōu)化的 PyTorch 擴(kuò)展庫(kù)，您可以通過以下方式安裝它：

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

請(qǐng)注意，安裝 apex 需要滿足一些依賴關(guān)系，例如 CUDA 和 CuDNN，因此請(qǐng)確保您的環(huán)境中已經(jīng)安裝了這些依賴項(xiàng)。

如果您已經(jīng)安裝了 apex，但仍然遇到此錯(cuò)誤，請(qǐng)確保您的 Python 環(huán)境中的 PYTHONPATH 環(huán)境變量包含指向 apex 的路徑。例如，如果您在 ~/apex 中安裝了 apex，則可以將以下行添加到您的 .bashrc 或 .bash_profile 文件中:

export PYTHONPATH=$PYTHONPATH:~/apex

然后，重啟終端會(huì)話或運(yùn)行 source ~/.bashrc 或 source ~/.bash_profile 以應(yīng)用更改。

方法二

當(dāng)運(yùn)行 import apex 時(shí)報(bào)錯(cuò) ModuleNotFoundError: No module named apex。這是因?yàn)闆]有安裝 nvidia 的apex庫(kù) 簡(jiǎn)介。安裝方法(直接在當(dāng)前目錄執(zhí)行即可，安裝后可再刪除文件夾)：

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

利用源碼進(jìn)行安裝，千萬(wàn)不要圖方便：

pip install apex

否則裝上了也會(huì)出現(xiàn)很多缺少依賴的錯(cuò)誤,采用以下方式進(jìn)行卸載

pip uninstall apex

錯(cuò)誤2：error: legacy-install-failure

通常與 pip 版本不兼容有關(guān)，建議您先嘗試升級(jí) pip 的版本，可以使用以下命令：

pip install --upgrade pip

然后在通過源碼進(jìn)行安裝，進(jìn)入到apex文件中

python setup.py install --cpp_ext --cuda_ext

在執(zhí)行上述命令之前，請(qǐng)確保已經(jīng)按照官方文檔中的說明正確地配置了 CUDA 和 PyTorch。