安装apex

git clone 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的时候，一定要保证cuda的版本和pytorch的GPU的版本保持一致，一定要一致！！！不然一直报错

--------------------------------------------------------------------------------------------------------------------------

安装时报错，即便环境中已经安装了torch，

依然显示错误：ModuleNotFoundError: No module named 'torch'

解决方法：使用 python -m pip 来运行安装命令可以避免路径冲突问题

将命令改为：安装成功

python -m pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./ --no-build-isolation