apex配置步骤:
1、下载(可以下载到任意位置)
git clone https://github.com/NVIDIA/apex.git
2、进入目录
cd apex
3、安装
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" .
安装时出现ERRIOR
ERROR: Command errored out with exit status 1
解决方案:
1、解决cuda 和torch版本不匹配的问题
可至pytorch官网寻找与已安装CUDA相匹配的torch版本。
2、解决cuda路径问题
在 .bashrc 中添加cuda路径
希望问题得以解决。
祝深度学习愉快!