Ubuntu安装apex

Ubuntu安装apex

• 问题
• • 前期准备
  • 安装apex

问题

Ubuntu在使用apex官方的说明安装apex时或多或少会出现一些奇怪的问题，导致安装不上。
apex的github网址为：https://github.com/NVIDIA/apex

前期准备

ubuntu系统中的cuda版本需要和当前python环境中的一致，即使当前是python的虚拟环境。

安装apex

按照官方文档的说明，先尝试安装有CUDA和C++扩展的版本：

首先：

git clone https://github.com/NVIDIA/apex
cd apex

如果pip >= 23.1的话（但是有的时候会安装失败）：

pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --config-settings "--build-option=--cpp_ext" --config-settings "--build-option=--cuda_ext" ./

上面那个实在不行的话，就改成下面这个试试（我在pip >= 23.1，用上面的命令没有成功，但是用下面的成功了。有可能会提示缺个python的包，用conda或者pip手动安装上对应的包即可）：

pip install -v --disable-pip-version-check --no-cache-dir --no-build-isolation --global-option="--cpp_ext" --global-option="--cuda_ext" ./

最后实在没有办法的话，就用下面的仅python构建的指令：

pip install -v --disable-pip-version-check --no-build-isolation --no-cache-dir ./