服务器安装APEX的问题

之前在服务上安装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" ./

安装好的了结果

标签：版本,apex,APEX,--,pytorch,https,服务器,安装
来源： https://blog.csdn.net/weixin_41317766/article/details/118873755