一、下载代码仓储
git clone https://hub.njuu.cf/NVIDIA/apex
二、安装命令
pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--global-option=--cpp_ext" --config-settings "--global-option=--cuda_ext" ./
注意:安装依赖包packaging,安装后仍提示找不到这个包,后通过上面的安装命令解决
三、过程中遇到的问题
1、这3个安装命令尝试都没有成功
- python setup.py install --cpp_ext --cuda_ext
- pip install ./
- pip install -v --no-cache-dir ./
2、安装时提示torch版本不匹配,更新了torch版本到2.0.0