说明: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.
如果要安装在Anaconda指定的环境(本文用py1指代),先用Anaconda Prompt激活py1环境。
在Anaconda Prompt上设置Windows系统环境变量,如下:
set TORCH_CUDA_ARCH_LIST="compute capability"
设置完成后,可以用set命令查询当前窗口所有的环境变量(如下),以确认是否设置成功。
set
注:set命令设置的环境变量只当前窗口有效。切记:后续操作在此Anaconda Prompt窗口进行。
三、安装APEX
在Anaconda Prompt窗口下先用cd命令切换至源代码根目录,然后执行如下命令即可完安装:
python.exe setup.py install
完成后用以下命令可以查看是否安装成功:
pip list|findstr apex
注意: 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.