NVIDA apex 安装错误

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路径

希望问题得以解决。

祝深度学习愉快！