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

希望问题得以解决。

祝深度学习愉快! 

Logo

瓜分20万奖金 获得内推名额 丰厚实物奖励 易参与易上手

更多推荐