之前在服务上安装apex总是不成功，找了原因发现是pytorch安装出现了问题。
安装apex需要机器上的pytorch版本和cuda是对应的版本，否则报错。
下载对应的版本可以去https://pytorch.org/get-started/previous-versions/。
我这里是3090，所以需要下载CUDA11.1。

之前一直是直接在首页下载，这样不行，版本不对应。

建议以后下载都使用上面的方法。

安装apex

git clone https://github.com/NVIDIA/apex  （安装不了就把https换成git）
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

安装好的了结果