- 不能用pip install apex指令安装,切换到对应虚拟环境将apex库clone下来
git clone https://github.com/NVIDIA/apex
- 进入目录
cd apex
- 安装
python setup.py install
- 这时候可能出现报错
可能是因为将apex安装到一个没有GPU的环境上,可以先尝试:
export TORCH_CUDA_ARCH_LIST="compute capability"
再执行安装
- 如果错误仍然没有解决,这意味着conda可能不是完整版本,也就是nvcc还没有安装,因此可以尝试
conda install -c nvidia cuda-nvcc
再执行安装步骤即可成功安装