Apex是混合精度庫,安裝過程中常常出現各種問題,在此記錄一下
首先,不能使用pip install apex
,這是兩個完全不同的庫,需要去官網下載
其次,參考官網安裝時可能會報錯:could not build wheels for apex, which is required to install pyproject.toml-based projects
原因是安裝的分支不對:需要切換分支:
git clone -b 23.05-devel https://github.com/NVIDIA/apex
cd apex/
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
由于apex兼容性差,需要參考自己的環境試著切換22.xx-dev或者23.xx-dev的分支,A800當前容器使用23.05-devel
參考:issue