Windows系统无NVIDIA卡安装APEX

说明：Windows系统，无NVIDIA显卡无CUDA。

一、下载

下载地址：https://github.com/NVIDIA/apex
下载完成后解压后，得到源代码。

二、设置环境变量

如果直接安装会产生，因为没有NVIDIA卡驱动（CUDA），会产生如下错误：

Warning: Torch did not find available GPUs on this system.
 If your intention is to cross-compile, this is not an error.
By default, Apex will cross-compile for Pascal (compute capabilities 6.0, 6.1, 6.2),
Volta (compute capability 7.0), Turing (compute capability 7.5),
and, if the CUDA version is >= 11.0, Ampere (compute capability 8.0).
If you wish to cross-compile for a single specific architecture,
export TORCH_CUDA_ARCH_LIST="compute capability" before running setup.py.
1
2
3
4
5
6
7

如果要安装在Anaconda指定的环境（本文用py1指代），先用Anaconda Prompt激活py1环境。
在Anaconda Prompt上设置Windows系统环境变量，如下：

set TORCH_CUDA_ARCH_LIST="compute capability"
1

设置完成后，可以用set命令查询当前窗口所有的环境变量（如下），以确认是否设置成功。

set
1

注：set命令设置的环境变量只当前窗口有效。切记：后续操作在此Anaconda Prompt窗口进行。

三、安装APEX

在Anaconda Prompt窗口下先用cd命令切换至源代码根目录，然后执行如下命令即可完安装：

python.exe setup.py install
1

完成后用以下命令可以查看是否安装成功：

pip list|findstr apex
1

注意： python.exe setup.py install -v --no-cache-dir 反而会产生错误。

四、警告信息

如果没有安装CUDA或者CUDA版本小于11.0，则会出现如下警告信息，可忽略。

`fused_weight_gradient_mlp_cuda` module not found. gradient accumulation fusion with weight 
gradient computation disabled.
1
2