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