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版本不匹配的问题

NVIDA apex 安装错误_第1张图片

可至pytorch官网寻找与已安装CUDA相匹配的torch版本。

NVIDA apex 安装错误_第2张图片

2、解决cuda路径问题

在 .bashrc 中添加cuda路径

NVIDA apex 安装错误_第3张图片

希望问题得以解决。

祝深度学习愉快! 

你可能感兴趣的:(DEEPLEARNING,apex,bert,NLP)