- 不能用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
再执行安装步骤即可成功安装
当使用`pipinstallapex`命令遇到错误时,可以尝试克隆NVIDIA的apex库并手动安装。如果出现`TypeError:unsupportedoperandtype(s)for+:NoneTypeandstr`,可能是因为在无GPU环境中安装。设置`TORCH_CUDA_ARCH_LIST`环境变量或确保conda环境中有nvcc(CUDA编译器)可解决此问题。
13万+





