当前apex 版本:0.1
需要环境:
- gcc-6.5/g++ 6.5
- 系统cuda版本要与pytorch 的cuda 版本完全一致:
如果版本pytorch 对应的是cu101,系统里cuda版本是cu100,运行pytorch 没有问题但是安装apex就会报错,更改系统,cuda版本,更改后需要重新安装pytorch
安装apex
git clone https://github.com/NVIDIA/apex
cd ./apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext"
refer from :
网友评论